luogu P4933 大师 线性dp
#include<map> #include<queue> #include<time.h> #include<limits.h> #include<cmath> #include<ostream> #include<iterator> #include<set> #include<stack> #include<cstdio> #include<cstring> #include<iostream> #include<algorithm> using namespace std; #define rep_1(i,m,n) for(int i=m;i<=n;i++) #define mem(st) memset(st,0,sizeof st) int read() { int res=0,ch,flag=0; if((ch=getchar())=='-') //判断正负 flag=1; else if(ch>='0'&&ch<='9') //得到完整的数 res=ch-'0'; while((ch=getchar())>='0'&&ch<='9') res=res*10+ch-'0'; return flag?-res:res; } typedef long long ll; typedef pair<int,int> pii; typedef unsigned long long ull; typedef pair<double,double> pdd; const int inf = 0x3f3f3f3f; const int N=1010; const int mod=998244353; int a[N]; int f[N][20000*2+10]; int ans; int main() { int n; cin>>n; for(int i=1; i<=n; i++) cin>>a[i]; int p=20000; for(int i=1; i<=n; i++) { ans++;//统计只有一个数字的情况 //f[i,k]以i结尾 公差为k的数量 for(int j=1; j<i; j++) { f[i][a[i]-a[j]+p]+=f[j][a[i]-a[j]+p]+1; f[i][a[i]-a[j]+p]%=mod; ans+=f[j][a[i]-a[j]+p]+1; ans%=mod; } } cout<<ans<<endl; return 0; }