tyvj 1091 问数列中的等差数列有多少个

	From Vivian Snow 等差数列

背景 Background     广东汕头聿怀初中 Train#3 Problem 3               描述 Description     等差数列的定义是一个数列S，它满足了(S[i]-S[i-1]) = d (i>1)。显然的一个单独的数字或者两个数字也可以形成一个等差数列。 经过一定的学习小C发现这个问题太简单了，等差数列的和不就是(Sn+S1)*n/2？因为这个问题实在是太简单了，小C不屑于去解决它。这让小C的老师愤怒了，他就找了另外一个问题来问他。 小C的老师给了他一个长度为N的数字序列，每个位置有一个整数，他需要小C帮他找到这个数字序列里面有多少个等差数列。 …… 这个问题似乎太难了，小C需要你的程序帮他来解决这个问题。               输入格式 Input Format     第一行一个整数N，表示老师给出的数字序列的长度。 第二行有N个整数A[i]，表示数字序列每个数字的大小。               输出格式 Output Format     输出只有一行一个整数，表示这个序列中的等差数列的个数（mod 9901）。               样例输入 Sample Input     5 1 4 2 3 7               样例输出 Sample Output     17               时间限制 Time Limitation     1s               注释 Hint     对于30%的数据，N <= 100 对于70%的数据，N <= 500 对于100%的数据，N <= 1000；-500 <= A[i] <= 500

 

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int data[1005],f[1005][2010];
//f[i][j]表示等差数列最后一项为i，数列公差为j的等差数的个数
const int BLOCK=1010;
int main()
{
    int n;
    while(cin>>n)
    {
        memset(f,0,sizeof(f));
        for(int i=1;i<=n;i++) cin>>data[i];
        int cnt=0;
        for(int i=1;i<=n;i++)
        {
            cnt=(cnt+i)%9901;//题目要求要mod9901
            for(int j=i-1;j>=1;j--)
            {
                int x=data[i]-data[j];//公差
                cnt=(cnt+f[j][x+BLOCK])%9901;
                f[i][x+BLOCK]=(f[i][x+BLOCK]+f[j][x+BLOCK]+1)%9901;
            }
        }
        cout<<cnt<<endl;
    }
    return 0;
}