P4933 大师

Aimee

很好想的是\(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;
}