很好想的是(f_{i,j})表示到第i个时公差为j,显然j不需要枚举,因为无论如何都会枚举i之前的电塔的,那样的话知道了电塔也就知道的公差了
转移的时候由于边界不太好考虑,那就先不考虑单元素了
所以从2开始枚举然后边枚举边统计,最后加上n
#include<iostream>
#include<cstdio>
#include<cstring>
#define int long long
using namespace std;
int n;
int h[1001];
int dp[1001][50001];
int ans;
int znx;
int mod=998244353;
signed main(){
scanf("%lld",&n);
for(int i=1;i<=n;++i){
scanf("%lld",&h[i]);
znx=max(h[i],znx);
}
for(int i=2;i<=n;++i){
int j;
for(int k=1;k<i;++k){
j=h[i]-h[k]+znx;
dp[i][j]=(dp[i][j]+dp[k][j]+1)%mod;
ans=(ans+dp[k][j]+1)%mod;
}
}
cout<<(ans+n)%mod;
return 0;
}