BZOJ1925: [Sdoi2010]地精部落 组合数+DP

BZOJ1925 [SDOI2010]地精部落

---

题意

给出$n$，求出$n$的全排列中，波动数列的个数$(0<n<4201)$

• 波动数列：每个点不是波峰（(比相邻的点都大）就是波谷（比相邻的点都小）

---

思路

• 最高峰的两边一定也是波动数列
  当最高峰$i$确定的时候：
  一个长度为$n$的波动数列可以由一个长度为$i-1$和长度为$n-1-i$的波动数列组合成
  一个波动数列由哪些数构成并没有关系（离散化后都一样）

那么我们只需要枚举最高峰的位置即可
波动数列的最高峰必定全在奇数位或者都在偶数位

• 波动数列对称后即为将最高峰在奇数位和在偶数位交换
  如：$3,1,4,2$对称后为$2,4,1,3$
  最高峰在奇数的情况数和最高位在偶数的情况数相同

状态

$dp[i]$表示长度为$i$的波动数列个数

状态转移方程

我们枚举最高峰位置为$j$,$comb[n][m]$表示组合数$n$选$m$
$$dp[i]=\sum _{j=0}^{n-1}dp[j]\ast dp[n-1-j]\ast comb[n-1][j]$$

长度为$j$的波动数列个数乘长度为$n-1-j$的波动数列个数再乘从$n-1$个数中选$j$个数的情况即可

---

代码

$dp$

LL dp[maxn],comb[maxn][2];
	for (int i = 2; i <= n; i++) {
		comb[0][i & 1] = comb[i][i & 1] = 1;//初始化组合数C[n][0]=C[n][n]=1
		for (int j = 1; j < i; j++) 
		comb[j][i & 1] = (comb[j - 1][i & 1 ^ 1] + comb[j][i & 1 ^ 1]) % mod;
		//杨辉三角形comb[i][j]=comb[i-1][j-1]+comb[i-1][j]
		//滚动数组优化组合数空间，i&1表示当前的组合数，i&1^1为上一个组合数
		for (int j = 0; j < i; j += 2)
        dp[i] = (dp[i] + dp[j] * dp[i - 1 - j] % mod * comb[j][i & 1 ^ 1] % mod) % mod;
        //状态转移方程，表示最高峰在奇数位的时候
	}
	cout << (dp[n] << 1) % mod << '\n';
	//对称要翻倍

AC

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn = 4205;
LL dp[maxn],comb[maxn][2];
int main() {
	ios::sync_with_stdio(false);
	int n; LL mod;
	cin >> n >> mod;
	dp[0] = dp[1] = 1;
	comb[0][0] = comb[0][1] = comb[1][1] = 1;
	for (int i = 2; i <= n; i++) {
		comb[0][i & 1] = comb[i][i & 1] = 1;
		for (int j = 1; j < i; j++) comb[j][i & 1] = (comb[j - 1][i & 1 ^ 1] + comb[j][i & 1 ^ 1]) % mod;
		for (int j = 0; j < i; j += 2) dp[i] = (dp[i] + dp[j] * dp[i - 1 - j] % mod * comb[j][i & 1 ^ 1] % mod) % mod;
	}
	cout << (dp[n] << 1) % mod << '\n';
}