洛谷 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][40010];
inline int read(){
    int k=0,f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    return k*f;
}
inline void MOD(int &k){if(k>=Mod) k-=Mod;}
int main(){
    n=read();
    for(rg int i=1;i<=n;i++) h[i]=read();
    for(rg int i=1;i<=n;i++){
        ans++;
        for(rg int j=i-1;j;j--){
            f[i][h[i]-h[j]+P]+=f[j][h[i]-h[j]+P]+1; MOD(f[i][h[i]-h[j]+P]);
            ans+=f[j][h[i]-h[j]+P]+1; MOD(ans);
        }
    }
    printf("%d
",ans);
    return 0;
}