波动数列(DP)

观察这个数列：

1 3 0 2 -1 1 -2 …

这个数列中后一项总是比前一项增加2或者减少3，且每一项都为整数。

栋栋对这种数列很好奇，他想知道长度为 n 和为 s 而且后一项总是比前一项增加 a 或者减少 b 的整数数列可能有多少种呢？

输入格式
共一行，包含四个整数 n,s,a,b，含义如前面所述。

输出格式
共一行，包含一个整数，表示满足条件的方案数。

由于这个数很大，请输出方案数除以 100000007 的余数。

数据范围
1≤n≤1000,
−109≤s≤109,
1≤a,b≤106
输入样例：
4 10 2 3
输出样例：
2
样例解释
两个满足条件的数列分别是2 4 1 3和7 4 1 -2。

分析
设第一个数为xx，则第二个数为x+d1x+d1，第三个数为x+d1+d2x+d1+d2 …。这里的d1d1，d2d2表示aa或者−b−b，所以这个数列为：

xx, x+d1x+d1, x+d1+d2x+d1+d2, x+d1+d2+d3x+d1+d2+d3, …, x+d1+d2+…+dn−1x+d1+d2+…+dn−1，又因为数列之和为s，所以转化成：

n∗x+(n−1)∗d1+(n−2)∗d2+(n−3)∗d3+…+dn−1=sn∗x+(n−1)∗d1+(n−2)∗d2+(n−3)∗d3+…+dn−1=s，再在一步转化：

s−[(n−1)∗d1+(n−2)∗d2+(n−3)∗d3+…+dn−1]n=xs−[(n−1)∗d1+(n−2)∗d2+(n−3)∗d3+…+dn−1]n=x
因为x是任意整数，所以又转化成：

ss与(n−1)∗d1+(n−2)∗d2+(n−3)∗d3+…+dn−1(n−1)∗d1+(n−2)∗d2+(n−3)∗d3+…+dn−1 模xx的余数相同。

到这里就转化成了组合问题。

下面就可以用闫氏dp分析法了。

1.状态表示：f[i][j]表示第i个选a或者-b且余数为j的所有集合的数量。
2.状态计算：第i个可以选a或者-b。

第i个选a：(n−1)∗d1+(n−2)∗d2+(n−3)∗d3+…+2∗dn−2+a模 x=jx=j。

则：(n−1)∗d1+(n−2)∗d2+(n−3)∗d3+…+2∗dn−2 模 x=j−a。

系数和下标之和为n，所以第i项的的系数为n-i。

所以：f[i][j] = f[i - 1][j - (n - i) * a]

第i个选b：同理：f[i][j] = f[i - 1][j + (n - i) * b]

#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1010, MOD = 100000007;

int f[N][N];

int get_mod(int a, int b)   // 求a除以b的正余数
{
    return (a % b + b) % b;
}

int main()
{
    int n, s, a, b;
    cin >> n >> s >> a >> b;

    f[0][0] = 1;
    for(int i=1;i<n;i++){
        for(int j=0;j<n;j++){
            f[i][j]=(f[i-1][get_mod(j-(n-i)*a,n)]+f[i-1][get_mod(j+(n-i)*b,n)])%MOD;
        }
    }
    cout << f[n - 1][get_mod(s,n)] << endl;

    return 0;
}