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

最新推荐文章于 2023-01-18 00:32:37 发布

cc 提升ing 变优秀ing

最新推荐文章于 2023-01-18 00:32:37 发布

阅读量226

点赞数 1

分类专栏：动态规划我认为的精华

本文链接：https://blog.csdn.net/weixin_42721412/article/details/107890742

版权

我认为的精华同时被 2 个专栏收录

195 篇文章 1 订阅

订阅专栏

动态规划

109 篇文章 0 订阅

订阅专栏

题目：

分析：

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

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

思路一：二维数组：

在这里插入图片描述

思路二：不用二维数组，直接循环求公差为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;
}