观察这个数列:
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。
做法:f[i][j]表示前i个的第j项的余数的值。 因为可以推出 下一项等于前一项c ± P ,P可能是a或者-b 所以可以推出 j = c - ai 或者 j = c + bi。 所以可得递推式:
f
[
i
]
[
j
]
=
(
f
[
i
−
1
]
[
g
e
t
m
o
d
(
j
+
a
∗
i
,
n
)
]
+
f
[
i
−
1
]
[
g
e
t
m
o
d
(
j
−
b
∗
i
,
n
)
]
)
f[i][j] = (f[i - 1][get_mod(j + a * i, n)] + f[i - 1][get_mod(j - b * i,n)]) % mod
f[i][j]=(f[i−1][getmod(j+a∗i,n)]+f[i−1][getmod(j−b∗i,n)])
#include <iostream>
using namespace std;
const int mod = 1e8 + 7;
int n,s,a,b;
int f[1005][1006];
int get_mod(int x, int y){
return (x % y + y) % y;
}
int main(){
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 + a * i, n)] + f[i - 1][get_mod(j - b * i,n)]) % mod;
}
}
cout << f[n - 1][get_mod(s,n)];
return 0;
}