校内互测 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<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<cmath>
#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);
}