[01背包]波动数列 2014年蓝桥杯

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

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

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

样例说明
这两个数列分别是2  4  1  3和7  4  1  -2。

输入

输入的第一行包含四个整数 n s a b，含义如前面说述。

数据规模和约定
对于100%的数据，1< =n< =1000，-1,000,000,000< =s< =1,000,000,000，1< =a,  b< =1,000,000。

输出

输出一行，包含一个整数，表示满足条件的方案数。由于这个数很大，请输出方案数除以100000007的余数。

样例输入

4 10 2 3

样例输出

2

题意：给出了波动数列的定义，询问在已知各参数的前提下有多少种不同的波动数列。

分析：经过转化后其实是个01背包问题，转化过程如下。

问题也就转化为已知x的值，求从{1, 2, ......, n-1}中选一组数来构成x的方案数，这个问题和01背包十分相似，自然也就能用01背包的思想解决，代码不多赘述，就是滚动数组优化01背包的模板，不过唯一需要注意的是第二层循环中j初值的设置，如果设置为n*(n-1)/2显然会超时，而从{1, 2, ......, i}中选数时和的最大值是i*(i+1)/2，因此j的初值设为i*(i+1)/2即可。

具体代码如下：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <string>
using namespace std;

int dp[1000005]; 
const int mod = 100000007;

signed main()
{
	int n, s, a, b;
	cin >> n >> s >> a >> b;
	dp[0] = 1;
	for(int i = 1; i <= n-1; i++)
		for(int j = i*(i+1)/2; j >= i; j--)
			dp[j] = (dp[j] + dp[j-i])%mod;
	long long ans = 0; 
	for(int x = 0; x <= n*(n-1)/2; x++)//根据x取值范围求解
		//首项a0 = (s+b*n*(n-1)/2-(a+b)*x)/n
		if((s+(long long)b*n*(n-1)/2-(a+b)*(long long)x)%n == 0)//首项a0存在 
			ans = (ans + dp[x])%mod;
	cout << ans << endl;
	
    return 0;
}