试题编号 | 201312-4 |
---|---|
试题名称 | 有趣的数 |
时间限制 | 1.0s |
内存限制 | 256.0MB |
问题描述 | 问题描述: 我们把一个数称为有趣的,当且仅当: 1. 它的数字只包含0, 1, 2, 3,且这四个数字都出现过至少一次。 2. 所有的0都出现在所有的1之前,而所有的2都出现在所有的3之前。 3. 最高位数字不为0。 因此,符合我们定义的最小的有趣的数是2013。除此以外,4位的有趣的数还有两个:2031和2301。 请计算恰好有n位的有趣的数的个数。由于答案可能非常大,只需要输出答案除以1000000007的余数。 输入格式 输入只有一行,包括恰好一个正整数n (4 ≤ n ≤ 1000)。 输出格式 输出只有一行,包括恰好n 位的整数中有趣的数的个数除以1000000007的余数。 样例输入 4 样例输出 3 |
考虑用动态规划来做。首先考虑数字摆放规律,按照题意,2在3的前面,0在1的前面,又0不能在首位,故首位必须是2。并且组成的数字必须包含0,1,2,3最少一次。假设前 i-1 位已经确定(必须含有2),现在考虑第 i 位数字有多少种可能(求出第i位数字有多少种可能也就解决了若位数为 i 时,有多少个有趣数)。有以下几种情况
- 前 i -1 位中有2,1,0 ,此时我们在第i位加上3即可。
- 前 i -1 位中有2,3,0,此时我们在第i位加上1即可。
- 前 i -1 位中全都有,此时我们在第 i 位可以加3 或 1(按照规则只能加这两种)。
我们将这三种情况分别标识为 3 4 5. 则
s
[
i
]
[
5
]
=
s
[
i
−
1
]
[
3
]
+
s
[
i
−
1
]
[
4
]
+
s
[
i
−
1
]
[
5
]
∗
2
s[i][5] = s[i-1][3] + s[i-1][4] + s[i-1][5] *2
s[i][5]=s[i−1][3]+s[i−1][4]+s[i−1][5]∗2
其中 s[i][j] 表示有 i 位数字时,数字组成为第j种状态有多少种可能。而我们的问题就是求解 s[n][5],即当数字有n位,并且0,1,2,3全部包含时有多少种可能。
继续考虑 s[i][3] 和 s[i][4] 的子问题:
- s[i][3]: (前 i -1 位中有2,1,0 )
- 前 i - 1位只包含2,1,不行。
- 前 i - 1位只包含2,0,可以,只需第 i 位为1即可
- 前 i - 1位只包含2,3,不行。
- s[i][4]: (前 i -1 位中有2,3,0)
- 前 i - 1位只包含2,1,不行。
- 前 i - 1位只包含2,0,可以,只需第 i 位为3即可
- 前 i - 1位只包含2,3,可以,只需第 i 位为0即可。
综上将2,0 表示为状态1,将2,3标识为状态2
最后状态1和2的子问题明显是前 i - 1 位只包含2。将其作为状态0。且易知 s[i][0] = 1.
下面给出代码(100分):
public class Main {
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
int num = scan.nextInt();
long[][] state = new long[num + 1][6];
long MOD = 1000000007;
// 初始化
for (int i = 0; i < 6; i++) {
state[0][i] = 0;
}
// 0 : 2
// 1 : 2,0
// 2 : 2,3
// 3: 2,0,1
// 4: 2,0,3
// 5: 都有
for (int i = 1; i <= num; i++) {
int j = i - 1;
state[i][0] = 1;
state[i][1] = (state[j][0] + state[j][1] * 2) % MOD;
state[i][2] = (state[j][0] + state[j][2]) % MOD;
state[i][3] = (state[j][1] + state[j][3] * 2) % MOD;
state[i][4] = (state[j][1] + state[j][2] + state[j][4] * 2) % MOD;
state[i][5] = (state[j][3] + state[j][4] + state[j][5] * 2) % MOD;
}
System.out.println(state[num][5]);
}
}