蓝桥杯题目练习 [蓝桥杯]波动数列

[蓝桥杯]波动数列

题目描述

观察这个数列：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;
}