给定一个数列,求其中所有等差数列个数(包含1、2个元素)
1.子状态:从左往右枚举,每次以新出现元素为结尾新增的等差数列个数
2.状态:dp[i][j]表示以第i个数为结尾,公差为j的数列的个数(元素个数大于等于2)(找一个角度来划分目标)
3.递推公式:dp[i][j]+=(dp[k][j]+1)(k=1.2.....i-1) (这里的+1是关键的地方,体现在若前面的数没有再前面的数则+1(做两个元素的等差数列)若有 则表示额外多了一个元素多的个数,而个数本身是由两个两个叠加起来的。这里计数注意只加ij部分(加的前面部分呢)
4.去重
#include<bits/stdc++.h>
using namespace std;
int dp[1005][40099];
const int mod=998244353;
int main()
{
int n;
int h[1005];
int N=20005;
cin>>n;
for(int i=1;i<=n;++i)
cin>>h[i];
long long ans=0;
for(int i=1;i<=n;++i)
{
ans=(ans+1)%mod;
for(int j=i-1;j>=1;--j)
{
int d=h[i]-h[j]+N;
dp[i][d]=(dp[i][d]+dp[j][d]+1)%mod;
ans=(ans+dp[j][d]+1)%mod;
}
}
cout<<ans;
return 0;
}