P4933 大师

原题链接

题解

对于任意剩余塔，都可以表示为以某个塔结尾的等差数列

code

#include<bits/stdc++.h>
using namespace std;
int h[1005]={0};
int dp[1005][40005]={0};//代表以塔i结尾，等差为j的种类
int haxi(int x)
{
    return x+20001;
}
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>h[i];

    int ans=0;
    for(int i=1;i<=n;i++)
    {
        ans+=1;//代表只有自己
        for(int j=1;j<i;j++) //这里j不能等于i，因为d为零的不止只有自己
        {
            int d=haxi(h[i]-h[j]);
            dp[i][d]=(dp[i][d]+dp[j][d]+1)%998244353;//这里要加不是赋值！
            ans+=(dp[j][d]+1);
            ans%=998244353;
        }
    }
    cout<<ans<<endl;
    return 0;
}