这题考动态规划
动态规划:由已知的状态推出未知的状态。
代码中的数组 status[ i ][ j ],表示第i位,第j个状态的个数。然后每一位数状态依赖于前一位的状态。
变化的量有两个:位数,状态数
位数为n
这题一共有6种状态
0 – 0 1 (2) 3
1 – (0) 1 (2) 3
2 – 0 1 (2) (3)
3 – (0) (1) (2) 3
4 – (0) 1 (2) (3)
5 – (0) (1) (2) (3)
括号括住的数表示这个状态的数字只包含的元素。
发现每个状态都有2,推导的出数字的第一位一定是2,所以每个状态下的数字都包含2。
为什么会得出这6种状态呢?下面的这篇博客讲的很好。
https://blog.csdn.net/u013580497/article/details/48326879
文章中的6个步骤其实就是对应上面文章说的6个状态。
由只含有一种元素的数字推出含有全部元素的数字个数。
当前位数的状态:
状态0:1
状态1:由前一位的状态0,1推出
状态2:由前一位的状态0,2推出
状态3:由前一位的状态1,3推出
状态4:由前一位的状态1,2,4推出
状态5:由前一位的状态3,4,5推出
1.由于第一位肯定为2,所以状态0肯定只包含2
2.状态1跟2,由于前面选了2,后面要只包含两个元素的数字,只能是(2,0)和(2,3),这就是状态1和2
3.状态3,4,都由状态1得出来的,由于状态1为(2,0),所以跟着后面的数字可以是1和3。所以(2,0,1),(2,0,3)为状态4,5
4.状态2后面有0,1,只有0可以接上,1不行,因为接了1,后面就不能接0了,0要在1左边。所以最后得出状态(2,3,0)和状态4一样,结果并入状态4。因此状态4由状态1和2得出。
5.状态5,补充最后一个数字。由状态3和4推出。
因为状态5是包含所有的数字,最后就输出第n位的状态5 --> status[n - 1][5]
java代码:
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner input = new Scanner(System.in);
int n = input.nextInt();
long[][] status = new long[n][6];
int mod = 1000000007;
status[0][0] = 1;
for (int i = 1; i < n; i++) {
status[i][0] = 1;
status[i][1] = (status[i - 1][1] * 2 + status[i - 1][0]) % mod;
status[i][2] = (status[i - 1][2] + status[i - 1][0]) % mod;
status[i][3] = (status[i - 1][3] * 2 + status[i - 1][1]) % mod;
status[i][4] = (status[i - 1][4] * 2 + status[i - 1][2] + status[i - 1][1]) % mod;
status[i][5] = (status[i - 1][5] * 2 + status[i - 1][4] + status[i - 1][3]) % mod;
}
System.out.println(status[n - 1][5]);
}
}
python代码:
n = int(input())
status = [[0] * 6 for _ in range(n)]
status[0][0] = 1
mod = 1000000007
for i in range(1, n):
status[i][0] = 1
status[i][1] = (status[i - 1][1] * 2 + status[i - 1][0]) % mod
status[i][2] = (status[i - 1][2] + status[i - 1][0]) % mod
status[i][3] = (status[i - 1][3] * 2 + status[i - 1][1]) % mod
status[i][4] = (status[i - 1][4] * 2 + status[i - 1][2] + status[i - 1][1]) % mod
status[i][5] = (status[i - 1][5] * 2 + status[i - 1][4] + status[i - 1][3]) % mod
print(status[n - 1][5])