历届试题 波动数列

  历届试题 波动数列  

时间限制：1.0s   内存限制：256.0MB

    

问题描述

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

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

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

输入格式

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

输出格式

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

样例输入

4 10 2 3

样例输出

2

样例说明

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

数据规模和约定

　　对于10%的数据，1<=n<=5，0<=s<=5，1<=a,b<=5；
　　对于30%的数据，1<=n<=30，0<=s<=30，1<=a,b<=30；
　　对于50%的数据，1<=n<=50，0<=s<=50，1<=a,b<=50；
　　对于70%的数据，1<=n<=100，0<=s<=500，1<=a, b<=50；
　　对于100%的数据，1<=n<=1000，-1,000,000,000<=s<=1,000,000,000，1<=a, b<=1,000,000。

 

一个很神奇的动态规划题目，至今没有完全AC；

分析：

http://blog.csdn.net/wr132/article/details/43861145

#include <cstdio>
#include <cstring>
#include <iostream>
#define mod 100000007
using namespace std;

int sum;
long long n,a,b,s;
int num[2][1010*1010];

void dp()
{
    memset(num,0,sizeof(num));
    
    num[0][0]=1;num[1][0]=1;
    
    for(int i=2;i<=n;i++)
    {
    	int k=i%2;
    	for(int j=1;j<=(n*(n-1))/2;j++)
    	{
    		if(i-1>j)
    		{
    			num[k][j]=num[1-k][j];
    		}
    		else
    		{
    			num[k][j]=(num[1-k][j-(i-1)]+num[1-k][j])%mod;
    		}
    	}
    }
}

int main()
{
    cin>>n>>s>>a>>b;
    
    dp();
    
    for(int i=0;i<=(n*(n-1))/2;i++)
	{
		long long x=s-(i*a)+(n*(n-1)/2-i)*b;
		if(x%n==0)
		{
			sum=(sum+num[n%2][i])%mod;
		}
	}
    
    printf("%d",sum);
    return 0;
}