校内互测 A. 等差数列 (dp)

 A. 等差数列
时间限制:1s 内存限制:128MB
题目描述
他还记得高一那年,他学习了等差数列。“如果一个数列{an}, n∈Z*, 满足对于任意i∈(1,n),有ai-ai-1=ai+1-ai,
则称{an}为一个等差数列。特别的,当 n≤2 时就必然是一个等差数列。”
比如6 6 6,就是一个典型的等差数列;而2 3 3则不是。
但是作为一名OIer,这对于他来说太简单了。他想要知道一个长度为n 的序列中,有多少个子序列是等差数列。子序列的意思是从整个序列中选出若干(可以不连续)的元素来,但是他们的顺序是不能改变的,且不能是空集。所以显然,一共会有2^n-1 个子序列,这实在太多了,所以你只需要输出答案对 9901 取模的结果就可以了。
输入格式
第一行一个正整数n,接下来一行n个数表示序列。
输出格式
一行一个数,表示答案。
输入样例
5
1 4 2 3 7

输出样例
17
数据范围
对于%10的数据,n≤3
对于%30的数据,n≤20
另外%30的数据,max{ai}-min{ai}≤1
对于%100的数据,n≤1000,-500≤ai≤500


题解:dp

f[i][j]表示以i结尾公差为j的等差数列的个数。

转移的时候枚举i,j,计算出a[j]-a[i],并用f[i][a[j]-a[i]]来更新f[j][a[j]-a[i]]。注意公差整体+1000,因为有可能为负数。

时间复杂度O(n^2)

#include
#include
#include
#include
#include
#include
#define N 2003
#define p 9901
using namespace std;
int n,a[N],ans,f[N][N];
int main()
{
   freopen("a.in","r",stdin);
   freopen("my.out","w",stdout);
   scanf("%d",&n);
   for (int i=1;i<=n;i++)  scanf("%d",&a[i]);
   memset(f,0,sizeof(f));
   for (int i=1;i<=n-1;i++)
    for (int j=i+1;j<=n;j++)
	 {
	 	int t=a[j]-a[i]+1000;
	 	f[j][t]=(f[j][t]+1)%p;
	 }
   for (int i=1;i<=n-1;i++)
   	for (int k=i+1;k<=n;k++)
   	{
	   int t=a[k]-a[i]+1000;
   	   f[k][t]=(f[k][t]+f[i][t])%p;
    }
	for (int i=1;i<=n;i++)
	 for (int j=0;j<=2000;j++)
	  ans=(ans+f[i][j])%p;
	ans=(ans+n)%p;
	printf("%d\n",ans);
}



你可能感兴趣的:(动态规划)