观察这个数列:
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;
}