JoyOI 1091 等差数列（DP）

题意分析

等差数列的定义是一个数列S，它满足了(S[i]-S[i-1]) = d (i>1)。显然的一个单独的数字或者两个数字也可以形成一个等差数列。
经过一定的学习小C发现这个问题太简单了，等差数列的和不就是(Sn+S1)*n/2？因为这个问题实在是太简单了，小C不屑于去解决它。这让小C的老师愤怒了，他就找了另外一个问题来问他。
小C的老师给了他一个长度为N的数字序列，每个位置有一个整数，他需要小C帮他找到这个数字序列里面有多少个等差数列。

一开始思维定势想的转移，dp[i][j][k]表示以i开头j结尾的序列，公差为k的序列个数，然后发现状态转移方程并不好写。

然后进行了一点改进dp[i][j]表示以i结尾公差为j序列的个数。 但是我是用的三层循环

for i:1 to n
    for j:i+1 to n
        for k:-500 to 500
            if(a[j] - a[i] == k) dp[j][k] += dp[i][k] +1

其实明显可以看出来第三层循环没有什么必要。因为枚举一对a[i]，a[j]，他们两个之间的差是一定的，没必要对于这一对数字枚举公差。

再说了$n^3$也是过不了的。

所以有了以下改进，状态的表述不变，改编成如下的枚举方法。

for i: 1 to n
    for j: i+1 to n
        dis = a[j] - a[i]
        dp[j][dis] += dp[i][dis] +1

还有一点需要注意注意的是，题目的数字范围是从 -500到500，数组索引不能为负数，所以需要有一个偏移量shift来控制。 详情请见代码。

代码总览

#include<bits/stdc++.h>
using namespace std;
const int nmax = 1005;
const int shift = 1000;
const int mod = 9901;
int a[nmax];
int dp[nmax][2050];
int n;
int main(){
    while(scanf("%d",&n)!= EOF){
        memset(dp,0,sizeof dp);
        memset(a,0,sizeof a);
        int ans = 0;
        for(int i = 1;i<=n;++i) scanf("%d",&a[i]);
        for(int i = 1;i<=n;++i){
            for(int j = i+1;j<=n;++j){
                int dis = a[j] - a[i];
                dp[j][dis+shift] += dp[i][dis+shift] +1;
                ans = (ans+dp[i][dis+shift] +1) % mod;
            }
        }
        printf("%d\n",ans+n);
    }
    return 0;
}