AcWing 1214. 波动数列

原题链接：https://www.acwing.com/problem/content/1216/

题目描述

观察这个数列：

1 3 0 2 -1 1 -2 …

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

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

输入格式

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

输出格式

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

由于这个数很大，请输出方案数除以1e8+7 的余数。

数据范围

1≤n≤1000,
−1e9≤s≤1e9,
1≤a,b≤1e6

输入样例：

4 10 2 3

输出样例：

2

 思路

假设满足条件波动数列的第一个数字取x，d(i)表示第i个项与前一项的差值，d(i)从+a和-b两个数中取

则该波动数列应为下列形式

x,  x+d(1),  x+d(1)+d(2),.....,x+d(1)+d(2)+...+d(n-1)

有又因为这n项的和为s

则                                    n*x+(n-1)*d(1)+(n-1)*d(2)+...+d(n-1)=s

整理可得

则由题意可以知道x 要是一个整数    则要使分子一定要是n的倍数

那么就可以知道

把多余的余数减去，这样才能保住分子是n的倍数

那么我们最后要求的就是这(n-1)个d的排列情况   然后找s%n的那个作为我们的答案

运用yxc的闫式dp分析法

假设f[i][j]表示    当前只考虑前i项，且当前的总和除以n的余数是j的方案的集合

在当前的f[i][j]下  我们取当前的项与前一项的差值为取+a 和 -b 两种情况

取+a时    f[i][j]+=f[i-1][(j-i*a)%n];      因为当前的d对后面的序列有影响 所以在当前就处理  同理

取-b时    f[i][j]+=f[i-1][(j+i*b)%n];

考虑边界情况   f[0][0]=1;     什么都没有的时候也算是一种选法

 代码

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;
const int MOD = 1e8+7;
const int N = 1010;
int f[N][N];//f[i][j]表示当前只考虑前i项，且当前的总和除以n的余数是j的方案的集合
int n, s, a, b;

int get_mod(int a,int b)//求a/b正余数    因为c++会出现负余数   但是我们通常定义的余数是正余数   //所以手动处理一下
{
    return (a % b + b) % b;
}

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 - i * a, n)]+f[i - 1][get_mod(j + i * b, n)])%MOD;
        }
    }
    
    cout << f[n-1][ get_mod(s, n)]<<endl;
}