原题链接: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;
}