洛谷：P4933 大师（dp，等差数列）

题目：

分析：

自己不适合看题解，看不懂，都是自己少看一点然后往那方面想。

看到等差就怕了，这个题很简单：

思路一：二维数组：

思路二：不用二维数组，直接循环求公差为0，1，2.。。。。的等差数列的个数，每一轮之间没有关系。我是用一个long long B[40005];//存放个数 表示以i这个数（而非i下标）结尾的个数。 相当于一个map。

代码：

#include<bits/stdc++.h>
using namespace std;
int m;
int A[1005];//存放原始数据 
long long B[40005];//存放个数   表示以i这个数结尾的个数。 
int main()
{
 cin>>m;
 int maxx=-1;
 int minn=1<<30;
 for(int i=0;i<m;i++) 
 {
  cin>>A[i];
  maxx=max(maxx,A[i]);
  minn=min(minn,A[i]);
 }
 long long ans=m;
 int t=maxx-minn;
 for(int i=1;i<=t;i++)
 {
  //cout<<ans<<"----"<<endl; 
  memset(B,0,sizeof(B));
  for(int ii=0;ii<m;ii++)
  {
   ans=(ans+B[A[ii]+i])%998244353;
   B[A[ii]]=(B[A[ii]]+B[A[ii]+i]+1)%998244353;
  }
  //cout<<ans<<endl;
  memset(B,0,sizeof(B));
  for(int ii=0;ii<m;ii++)
  {
   if(A[ii]-i>=0)
   {
   ans=(ans+B[A[ii]-i])%998244353;
   B[A[ii]]=(B[A[ii]]+B[A[ii]-i]+1)%998244353;
   }
   else{
    B[A[ii]]++;
   }
  }
  //cout<<ans<<endl;
 }
 memset(B,0,sizeof(B));
  for(int ii=0;ii<m;ii++)
  {
   ans=(ans+B[A[ii]])%998244353;
   B[A[ii]]=(B[A[ii]]+B[A[ii]+0]+1)%998244353;
  }
 cout<<ans;
}