[蓝桥杯]波动数列
题目描述
观察这个数列:1 3 0 2 -1 1 -2 …这个数列中后一项总是比前一项增加2或者减少3。
栋栋对这种数列很好奇,他想知道长度为 n 和为 s 而且后一项总是比前一项增加a或者减少b的整数数列可能有多少种呢?
输入
输入的第一行包含四个整数 n s a b,含义如前面说述。
1<=n<=1000,-1,000,000,000<=s<=1,000,000,000,1<=a, b<=1,000,000。
输出
输出一行,包含一个整数,表示满足条件的方案数。由于这个数很大,请输出方案数除以100000007的余数。
样例输入 Copy
4 10 2 3
样例输出 Copy
2
来自大佬的代码:https://www.cnblogs.com/imzscilovecode/p/8666783.html
#include<iostream>
#include<string>
#include<algorithm>
#include<vector>
#include<queue>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int maxn = 1000000+10;
const int mod = 100000000+7;
ll dp[500000+10];
int main(){
ll n,s,a,b;
cin>>n>>s>>a>>b;
dp[0]=1;
for(int i=1;i<n;i++){
for(int j=i*(i+1)/2;j>=i;j--){
dp[j]=(dp[j-i]+dp[j])%mod;
}
}
ll div=(n-1)*n/2;
ll sum=s+div*b;
ll dir=a+b;
ll ans=0;
for(int i=0;i<=div;i++){
if(sum%n==0){
ans=(ans+dp[i])%mod;
}
sum-=dir;
}
cout<<ans;
return 0;
}