波动数列(数论推导,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。

算法分析

和s == x + (x+d1)+(x+d1+d2)+…+(x+d1+…dn-1) 其中d∈{a,-b};
整理式子得 s == n * x + (n-1) * d1 +(n-2) * d2+…+dn-1;
s给了,n给了,a b给了所以d也是知道的,那么就要求x的可能数了,我们继续转化
x == {s -((n-1) * d1 +(n-2) *d2+…+dn-1)} /n;
那么x是整数,所以{s -((n-1) * d1 +(n-2) * d2+…+dn-1)} / n得到的是整数就意味着
s -((n-1) * d1 +(n-2) * d2+…+dn-1)是n的倍数,所以两个分式余n的余数是要相同的,这样可以使n整除
s%n是直接知道的,那么我们就要求((n-1) * d1 +(n-2)*d2+…+dn-1)%n了.这些都是一步一步来的,所以我们用动态规划的思想来进行
dp[i][j]表示前i项的和余n的余数为j
我们定义sum为前i-1项的和,第i项为i * a
我们能得到 (sum+(i * a))%n == j .j是小于n的所以j%n == j这样我们转换为同余方程
同余方程是可以直接加减的
(sum+(i * a))%n == j%n =>sum%n == (j-(i * a))%n.
所以我们能得到状态转移方程dp[i][j]=dp[i-1][(j-(i * a))%n]+dp[i-1][(j+(i * b))%n];
其中是有负数的所以我们要保证余数为正,所以加上一个方程就行了(a%b+b)%b就可以保证余数为正

代码实现

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=1e3+5;
const int mod=100000007;
int dp[maxn][maxn];
int n,s,a,b;
int get_mod(int a,int b)
{
    return (a%b+b)%b;
}
int main()
{
    cin>>n>>s>>a>>b;
    dp[0][0]=1;
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<n;j++)
        {
            dp[i][j]=(dp[i-1][get_mod(j-a*i,n)]+dp[i-1][get_mod(j+b*i,n)])%mod;
        }
    }
    cout<<dp[n-1][get_mod(s,n)]<<endl;
    return 0;
}