摆动序列

摆动序列

问题描述
　　如果一个序列的奇数项都比前一项大，偶数项都比前一项小，则称为一个摆动序列。即 a[2i]<a[2i-1], a[2i+1]>a[2i]。
　　小明想知道，长度为 m，每个数都是 1 到 n 之间的正整数的摆动序列一共有多少个。
输入格式
　　输入一行包含两个整数 m，n。
输出格式
　　输出一个整数，表示答案。答案可能很大，请输出答案除以10000的余数。
样例输入
3 4
样例输出
14
样例说明
　　以下是符合要求的摆动序列：
　　2 1 2
　　2 1 3
　　2 1 4
　　3 1 2
　　3 1 3
　　3 1 4
　　3 2 3
　　3 2 4
　　4 1 2
　　4 1 3
　　4 1 4
　　4 2 3
　　4 2 4
　　4 3 4
评测用例规模与约定
　　对于 20% 的评测用例，1 <= n, m <= 5；
　　对于 50% 的评测用例，1 <= n, m <= 10；
　　对于 80% 的评测用例，1 <= n, m <= 100；
　　对于所有评测用例，1 <= n, m <= 1000。

一开始看到这道题，就想着dfs枚举所有结果，然后统计数量，但是这样会超时。

dfs代码：

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

int ans;
int m, n;

void dfs(vector<int> arr, int k) {
	//出口 
	if (k == m + 1) {
		ans = (ans + 1) % 10000;
		return;
	}
	if (k % 2 == 0) {//偶数项枚举比前一项小的
		for (int i = arr[k - 1] - 1; i > 0; i--) {
			arr[k] = i;
			dfs(arr, k + 1);
		}
	}
	else {奇数项枚举比前一项大的
		for (int i = arr[k - 1] + 1; i <= n; i++) {
			arr[k] = i;
			dfs(arr, k + 1);
		}
	}
}

int main() {
	cin >> m >> n;
	vector<int> arr(m + 1, 0);
	arr[0] = 0;
	dfs(arr, 1);
	cout << ans << endl;
	return 0;
}

深搜枚举行不通，就想着有无规律可循，所以就有了dp的解法
dp表如下，表达能力有限，说不清楚，动手画一次就可以懂了

dp解法代码：

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

int ans;
int m, n;
const int MOD = 10000;

int main() {
	cin >> m >> n;
	vector< vector<int> > dp(m + 1, vector<int>(n + 1, 0));
	//初始化第一行
	for (int j = 1; j <= n; j++)
		dp[1][j] = 1;
	for (int i = 2; i <= m; i++) {
		if (i & 1) {//奇数项
			for (int j = 2; j <= n; j++) {
				dp[i][j] = (dp[i][j - 1] + dp[i - 1][j - 1]) % MOD;
			}
		}
		else {//偶数项
			for (int j = n - 1; j >= 1; j--) {
				dp[i][j] = (dp[i][j + 1] + dp[i - 1][j + 1]) % MOD;
			}
		}
	}
	//求和
	for (int j = 1; j <= n; j++) {
		ans = (ans + dp[m][j]) % MOD;
	}
	cout << ans << endl;
	return 0;
}