洛谷 4933 洛谷10月月赛II T2 大师

【题解】

　　f[i][j]表示最后一个数为h[i]，公差为j的等差数列的个数。n方枚举最后一个数和倒数第二个数转移即可。注意公差可能为负数，需要移动为正数再作为下标。

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #define LL long long
 #define rg register
 #define N 1010
 #define Mod (998244353)
 #define P (20000)
 using namespace std;
 int n,mx,ans,h[N],f[N][];
 inline int read(){
     int k=,f=; char c=getchar();
     while(c<''||c>'')c=='-'&&(f=-),c=getchar();
     while(''<=c&&c<='')k=k*+c-'',c=getchar();
     return k*f;
 }
 inline void MOD(int &k){if(k>=Mod) k-=Mod;}
 int main(){
     n=read();
     for(rg int i=;i<=n;i++) h[i]=read();
     for(rg int i=;i<=n;i++){
         ans++;
         for(rg int j=i-;j;j--){
             f[i][h[i]-h[j]+P]+=f[j][h[i]-h[j]+P]+; MOD(f[i][h[i]-h[j]+P]);
             ans+=f[j][h[i]-h[j]+P]+; MOD(ans);
         }
     }
     printf("%d\n",ans);
     return ;
 }