蓝桥杯 2014 省赛 波动数列

原题题目

观察这个数列：

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。

思路及证明过程

 

 

最终代码

def get_mod(a, b):
    return (a % b + b) % b


if __name__ == '__main__':
    MOD = 100000007
    N = 1005
    dp = [[0] * N for _ in range(N)]
    n, s, a, b = map(int, input().split())
    dp[0][0] = 1
    for i in range(1, n):
        for j in range(0, n):
            dp[i][j] = (dp[i][j] + dp[i - 1][get_mod(j - i * a, n)]) % MOD
            dp[i][j] = (dp[i][j] + dp[i - 1][get_mod(j + i * b, n)]) % MOD
    print(dp[n - 1][get_mod(s, n)])