蓝桥杯学习记录||1214. 波动数列

AcWing||1214. 波动数列

活动地址：https://www.acwing.com/activity/content/19/

考察要点：DP

题目要求

观察这个数列：

1 3 0 2 -1 1 -2 …

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

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

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

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

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

数据范围

1≤n≤1000,
−10⁹≤s≤10⁹,
1≤a,b≤10⁶

输入样例：

4 10 2 3
输出样例：

2
样例解释

两个满足条件的数列分别是

2 4 1 3 和 7 4 1 -2。

题目地址：https://www.acwing.com/problem/content/1216/

解析：看起来像数列的问题，但是本质上还是背包问题，每次都有两个选择 加a 或 减b ，使用 f[i][j] 表示考虑前 i 个数，前 i 个数的总和 % n 为 j 的方案的集合

注意：c++取余可能出现负数，使用写一个函数求正余数：先让 a 对 b进行一次取余，结果的范围在 （-b，b）之间，+b后再次取余，就一定是正的了。

如果不理解，也可以在第一次取余后加一个判断如果为负数，就直接 加 b

 int getMod(int a, int b)  //求 a % b 的正余数
 {
     int x = a % b;
     if(x < 0) x += b;
     return x;
 } 

代码思路 先枚举前 i 个数和各种余数的情况，i 加上 i - 1 的各种结果。最后输出 余数 与 s % n相等的集合

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
 using namespace std;
 
 int n, s, a, b;
 const int N = 1010, MOD = 100000007;
 int f[N][N];   //只考虑前i项，且当前的总和%n为j的方案的集合
 
 int getMod(int a, int b)  //求 a % b 的正余数
 {
     return (a % b + b) % b;
 }
 
int main()
{
    cin >> n >> s >> a >> b;
    
    f[0][0] = 1;  // 【一个数都没取】【余数为0】= 一种方案
    
    //枚举所有状态
    for(int i = 1; i < n; i ++)
        for (int j = 0; j < n; j ++ )   //余数范围【0~n-1】
        f[i][j] = (f[i - 1][getMod(j - a *(n - i), n)] + f[i - 1][getMod(j + b *(n - i), n)]) % MOD;
        
    cout << f[n - 1][getMod(s, n)] << endl;
    
    return 0;
}

本题:1214. 波动数列
代码参考:视频讲解