【CCF-CSP】201312-4 有趣的数 (划分状态dp）

【CCF-CSP】201312-4 有趣的数

题目

给一个数 n，求长度为 n 且满足一下条件的数字串的个数，结果取模。

1. 只包含 0, 1, 2, 3，且必须包含
2. 0 不作为开头
3. 所有 0 在 所有 1 前面，所有 2 在所有 3 前面

分析

遇到这样的题，肯定要先写几项，观察规律。可以看出 2 一定开头，因为数字 1、3都有限制，而 0 又不能作为开头。不过好像没什么用。

在写规律的时候发现，长度为 n 的所有字符串都可以由长度为 n -1 的字符串添一个字符推出，有点 dp 的感觉。

不过添加字符的方式有很多，不好写转移方程。因此对于长度为 n 的集合，要继续划分更细的状态。首先注意到题目中很重要的限制：所有 0 在 所有 1 前面，所有 2 在所有 3 前面，因此状态转移方程中要根据最后一位加不加 1, 3来做（这样才会满足限制）。

最终肯定要求由且仅由 {0,1,2,3} 组成的字符串。那么相应的可以从 {0, 1, 2} 加 3，或者 {0, 2, 3} 加 1 来转移过来。

{0, 1, 2, 3} = {0, 1, 2} +3;
{0, 1, 2, 3} = {0, 2, 3} +1;

同理：

{0, 1, 2} = {0, 2} +1;
{0, 2, 3} = {0, 2} +3;
{0, 2} = {2} + 0;

通过上面的分析，可以很自然得出状态转移方程。$dp[i][j]$ 表示长度为 i，且状态为 j，的所有方案数。

#include <bits/stdc++.h>
using namespace std;
#define d(x) cout<<x<<endl

typedef long long ll;
const int INF = 0x3f3f3f3f;
const int N = 1e3 + 10;
const int mod = 1e9 + 7;

int n;
ll dp[N][6];

int main() {
	int n;
	scanf("%d", &n);
	for (int i = 1; i <= n; i++) {
		dp[i][1] = 1;
		// 下面 (* 2) 代表从上个状态加两种数字达到
		dp[i][2] = (dp[i - 1][2] * 2 + dp[i - 1][1]) % mod; 
		dp[i][3] = (dp[i - 1][3] + dp[i - 1][1]) % mod;
		dp[i][4] = (dp[i - 1][4] * 2 + dp[i - 1][2]) % mod;
		dp[i][5] = (dp[i - 1][5] * 2 + dp[i - 1][2] + dp[i - 1][3]) % mod;
		dp[i][6] = (dp[i - 1][6] * 2 + dp[i - 1][4] + dp[i - 1][5]) % mod;
	}
	printf("%lld\n", dp[n][6]);
	return 0;
}