P4933 大师

点击查看代码

#include<bits/stdc++.h>
using namespace std;

const int N=1010,MOD=998244353,OFFSET=2e4+5;
int h[N];
int n;
int f[N][40010];

int main()
{
    ios::sync_with_stdio(0),cin.tie(0);
    cin>>n;

    for(int i=1;i<=n;i++) cin>>h[i];

    int ans=n;

    for(int i=1;i<=n;i++){
        for(int j=1;j<i;j++){
            int d=h[i]-h[j]+OFFSET;

            f[i][d]=(f[i][d]+f[j][d]+1)%MOD;

            ans=(ans+f[j][d]+1)%MOD;
        }
    }

    cout<<ans<<"\n";

    return 0;

    
}