洛谷 P4933 大师（DP）

题目链接：https://www.luogu.com.cn/problem/P4933

 

方程的设计比较难想：

设f[i][j]表示等差数列的最后一个数的位置为i，公差为j的方案数。

 

转移的话：

枚举k从1到i，f[i][j]+=f[k][j]。

最后累加答案，注意ans+i的含义（一个及两个的方案数）。

 

AC代码：

#include<cstdio>
#include<iostream>
using namespace std;
const int N=1005;
const int M=20000;
const int mod=998244353;
int n,ans;
int h[N],f[N][M<<1];
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&h[i]);
    for(int i=1;i<=n;i++){
        ans=(ans+i)%mod;
        for(int j=1;j<i;j++){
            int now=h[i]-h[j];
            ans=(ans+f[j][now+M])%mod;
            f[i][now+M]=(f[i][now+M]+f[j][now+M]+1)%mod;
        }
    }
    printf("%d",ans);
    return 0;
}