波动数列（主题库3090）

题目描述

观察这个数列：

　　1 3 0 2 -1 1 -2 ...

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

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

输出格式

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

输出格式

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

样例

输入数据#1

4 10 2 3

输出数据#1

2

解释#1

这两个数列分别是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代码：

#include<iostream>
#include<cstdio>
using namespace std;

typedef long long ll;
const int MAX=5e5+2,mod=100000007;
ll ch[MAX]={0};
ll n,s,a,b;
ll a0,k,gcd;
ll gcdxy(ll n,ll m,ll &x,ll &y,ll sum){
	if(m){
		ll gcd=gcdxy(m,n%m,y,x,sum);
		y-=n/m*x;
		return gcd;
	}
	else{
		if(sum%n){
			x=y=0;
			return -1;
		}
		else{
			x=sum/n;
			y=0;
			return n;
		}
	}
}

int main(){
	scanf("%lld%lld%lld%lld",&n,&s,&a,&b);
	ch[0]=1;
	ll ind=0;
	for(int i=1;i<n;i++){
		ch[ind+i]+=ch[ind];
		for(int j=ind-1;j>=0;j--){
			if(ch[j]){
				ch[j+i]+=ch[j];
				ch[j+i]%=mod;
			}
		}
		ind=ind+i;
	}
	gcd=gcdxy(n,a+b,a0,k,s-n*(n-1)*a/2);
	if(gcd==-1){
		cout<<0<<endl;
		return 0;
	}
	k=-k;
	ll d=n/gcd;
	ll maxnum=n*(n-1)/2;
	ll sumnum=0;
	ind=(k%d+d)%d;
	for(int i=ind;i<=maxnum;i+=d){
		sumnum+=ch[i];
		sumnum%=mod;
	}
	printf("%lld\n",sumnum);
	return 0;
}

https://www.topscoding.com/p/3090https://www.topscoding.com/p/3090