波动数列（笔记）

题目链接.https://www.dotcpp.com/oj/problem1449.html

题干：

观察这个数列：
1 3 0 2 -1 1 -2 …

这个数列中后一项总是比前一项增加2或者减少3。

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

样例说明
这两个数列分别是2 4 1 3和7 4 1 -2。
数据规模和约定
对于100%的数据，1< =n< =1000，-1,000,000,000< =s< =1,000,000,000，1< =a, b< =1,000,000。

前段时间了解了一些关于二叉树的知识，发现这道题也可以用类似的方式画出一棵树

发现+a，-b在整个式子中会出现(n-1)*n/2次，那么就只用考虑+a,-b的占比分类计算就好，于是就出现了以下代码

#include <iostream>
using namespace std;

static double times;

int main()
{
    int n;
    double s,a,b;
    cin>>n>>s>>a>>b;
    double start=(int)(s/n-b-b);
    times=0;
    while(start*n+(n*(n-1)*a/2)>s)
        start--;
    double max=((int)s/n+a+a);
    while(max*n-(n*(n-1)*b/2)<s)
        max++;
    while(start<=max){
        for(int x=0;x<=(n-1)*n/2;x++){
            double sum=(start*n+a*x-b*((n-1)*n/2-x));
            if(sum==s)
                times++;
            else if(sum>s||(start*n+a*(n-1)*n/2)<s)
                x=(n-1)*n/2;
        }
        start++;
    }
    cout<<(int)times%100000007;
}

然而，很明显，这个程序算法效率很低

然后就去了解了一下闫氏dp分析法，然后继续做这道题😐

设s=start+(start+p1)+(start+p1+p2)+(start+p1+p2+p3)+…+(start+p1+p2+…+pn-1)
s=n∗start+(n−1)∗p1+(n−2)∗p2+(n−3)∗p3+…+pn−1
s%n=[(n−1)∗p1+(n−2)∗p2+(n−3)∗p3+…+pn−1]%n
容我先画个图

i：开始➕a的第i层
j：此时a的个数
dp[i][j]:代表序列前i时选择j个a的方案

首先考虑前2项的时候会有几个a，记为dp[0]=1,dp[1]=1，此时这个三角的可能性就考虑清楚了

其次考虑前3项，时我们将上面这个三角放在现在的三角的左下角就会有这样的结构现在我们其他的东西不清楚但是上面的三角形所得到的方案可以直接镶嵌到现在的最底层，这是没有问题的

这也就是为什么我们可以直接让j=i的地方开始考虑，因为在整个式子中取i以下个a的情况都可以包含在上一层里面，所以之后的所考虑的情况都从第1次就考虑+a的情况开始

按照公式，dp[2]=dp[0]+dp[2]=1,dp[3]=dp[1]+dp[3]=1,在选择第i个数的时候，如果不用+a的话，就直接继承上一轮的dp[j],如果需要就加上dp[j-i]，dp[j-i]中储存的是上一层应该选择的a的数量的方案数

因为如果有了第i层的时候，第2层选择的a就会在此多出i个a所以

dp[6]=dp[3]+dp[6]=1,dp[5]=dp[2]+dp[5]=1,dp[4]=dp[1]+dp[4]=1,dp[3]=dp[0]+dp[3]=2

所以状态计算就是dp[j]=dp[j-i]+dp[j]

🌰：4 10 2 3
//代码来自https://blog.csdn.net/SunMoonVocano/article/details/79464889

#include <iostream>
using namespace std;
typedef long long ll;
 
const ll mod = 100000007;
ll dp[10000006];
 
int main()
{
    ll n,s,a,b;
	cin>>n>>s>>a>>b;
	ll ans = 0;
	
	dp[0] = 1;
	for(ll i = 1; i < n; i++)
        for(ll j = i*(i+1)/2; j >= i; j--)
            dp[j] = (dp[j-i] + dp[j])%mod;
        
	for(ll i = 0; i <= (n-1)*n/2; i++)
	{
		ll tmp = s + i*a - ((n-1)*n/2 - i) * b;
		if(tmp % n == 0) ans = (ans+dp[i])%mod;  
	} 
	cout << ans <<endl;
    return 0;
}