【蓝桥杯】真题训练 2014年C++A组 题10 波动数列

波动数列

问题描述

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

题目解析

 枚举首项+深搜

方法一 10分

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

//枚举首项+深搜
int n, s, a, b; 
const int MOD = 100000007;
int res;

void dfs(int x, int cnt, int sum, vector<int> path)	{	//x上一个， cnt已经有n个，sum截止上一项已经求得和 
	if(cnt == n){
		if(sum == s){
			res ++;
	//		for(int i = 0; i < path.size(); i++){
	//			cout << path[i] << " "; 
	//		}
	//		cout << endl;
		}
		if(res > MOD){
			res %= MOD;
		}
		return;
	}
	path.push_back(x+a);
	dfs(x+a, cnt+1, sum+x+a, path) ;
	path.erase(path.end()-1);
	
	path.push_back(x-b);
	dfs(x-b, cnt+1, sum+x-b, path);	
	path.erase(path.end()-1);
}

int main(int argc, char** argv) {
	cin >> n >> s >> a >> b;
	int x; 
	//枚举首项 
	int x1 =  (s - a*n*(n-1)/2)/n;
	int x2 =  (s + b*n*(n-1)/2)/n;
	
	for(x = x1; x <= x2; x++){
		vector<int> path;
		path.push_back(x); 
		dfs(x, 1, x, path);
	}
	cout << res << endl;
	
	return 0;
}