蓝桥杯-历届试题-波动数列（C/C++）

观察这个数列：

1 3 0 2 -1 1 -2 …

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

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

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

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

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

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

动态规划 O(n^2)

#include <iostream>
#include <cstdio>
#include <cmath>

using namespace std;

const int N = 1010, mod = 100000007;

// 波动序列可以表示为 x , x + d1, x + d1 + d2, ..., x + d1 + ... + dn - 1，化简表示它们的和为s
// nx + (n - 1)d1 + (n - 2)d2 + ... + dn-1 = s (di可取 +a 或 -b)
// x = (s - ((n - 1)d1 + (n - 2)d2 + ... + dn - 1)) / n 有整数解，故问题就转化为
// 求满足多项式(n - 1)d1 + (n - 2)d2 + ... + dn - 1 模n同余 s 的方案数

int f[N][N];

//状态表示：f[i, j]表示确定了多项式前i项，多项式的值模n的余数为j的方案数
// f[i, j] = (f[i - 1, (j - (n - i) * a) % n] + f[i - 1, (j + (n - i) * b) % n]) % mod
// f[0, 0] = 1

int n, s, a, b;

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

int main(){
    scanf("%d %d %d %d", &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;
            }
        }
    printf("%d\n", f[n - 1][get_mod(s, n)]);
    return 0;
}

记忆化搜索

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>

using namespace std;

const int N = 1010, mod = 100000007;

int memo[N][N]; //memo[i, j]表示确定了多项式的前i项，其值模n的余数为j的方案数

int n, s, a, b;

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

int dfs(int i, int j){
    if(memo[i][j] != -1) return memo[i][j];
    if(i == 0){
        if(j == 0) return 1;
        return 0;
        }
    int sum = 0;
    sum = (sum + dfs(i - 1, get_mod(j - (n - i) * a, n))) % mod;
    sum = (sum + dfs(i - 1, get_mod(j + (n - i) * b, n))) % mod;
    return memo[i][j] = sum;
}

int main(){
    scanf("%d %d %d %d", &n, &s, &a, &b);
    memset(memo, -1, sizeof(memo));
    int ans = dfs(n - 1, get_mod(s, n));
    printf("%d\n", ans);
    return 0;
}