问题描述
如果一个序列的奇数项都比前一项大,偶数项都比前一项小,则称为一个摆动序列。即 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用一个记忆数组dp[i][j]维护一下长度为i最后一位是j的情况有多少种,但是这样优化还是比较慢,因为每次不能借用前一种的情况最后我们再想,可以采用二维dp的方法dp[i][j]跟上种方法类似,表示长度为i最后一位<=j(奇数是小于等于,偶数是大于等于)的情况有多少种然后每次的更新利用上一次更新做到优化最后可以使用O(n2)复杂度完成i和j对应的m和n,m和n都是1e3,所以足够开一个二维数组,然后开始找状态转移方程。
如果i是奇数,说明他要找比上一个数他小的情况就加上dp[i−1][j−1]让 j 从 n到1枚举,然后每次要加上一次更新的情况(dp[i][j+1]);如果i是偶数,就要找比一个数大的情况也就是dp[i−1][j+1]让j 从 1到n枚举,然后每次更新上一次更新的情况(dp[i][j−1]),所以可以找到状态方程:
dp[i][j]=(dp[i−1][j−1]+dp[i][j+1])%mod,i为奇数
dp[i][j]=(dp[i−1][j+1]+dp[i][j−1])%mod,i为偶数
//#include<bits/stdc++.h>
#include<iostream>
#define mod 10000
using namespace std;
int dp[1010][1010];
int main() {
// m为长度,n为数的最大取值范围
int m,n;
cin>>m>>n;
for(int i=1; i<=n;i++)
dp[1][i]=n-i+1;//初始化第一位为i的时候有n-i+1种情况
for(int i=2;i<=m;i++)
if(i&1)//奇数倒着更新
for(int j=n;j>=1;j--)
dp[i][j]=(dp[i-1][j-1]+dp[i][j+1])%mod;
else//偶数顺着更新
for(int j=1;j<=n;j++)
dp[i][j]=(dp[i-1][j+1]+dp[i][j-1])%mod;
int ans=m&1?dp[m][1]:dp[m][n];
cout<<ans;
return 0;
}