CCF 201312-4 有趣的数

又又又又又又又又是dp
解释一个应该就都懂了:
dp[i + 1][5] = (dp[i][5] * 2 + dp[i][3] + dp[i][4]) % mod;
含义为由2,0,1,3组成的i+1位数的组合可能性等于(由2,0,1,3组成的i位数末尾(即第i+1位)加上3或1,由2,0,1组成的i位数末尾(即第i+1位)加上3,由2,0,3组成的i位数末尾(即第i+1位)加上1)的三种状态之和 .

#include <cstring>
#include <string.h>
#include <iostream>
#include <stdio.h>
#include <unordered_map>
#include <map>
#define ll long long
using namespace std;
#define TLE ios::sync_with_stdio(0),cin.tie(0)
const int INF = 0x3f3f3f3f;
#define mod 1000000007
int main() {
	TLE;
	ll dp[1005][6] = { 0 }, n;
	cin >> n;
	dp[1][0] = 1;
	for (int i = 1; i < n; i++) {
		dp[i + 1][0] = 1 % mod; // 全部为2组成;
		dp[i + 1][1] = (dp[i][1] * 2 + dp[i][0]) % mod; // 为2,0组成;
		dp[i + 1][2] = (dp[i][2] + dp[i][0]) % mod; // 为2,3组成;
		dp[i + 1][3] = (dp[i][3] * 2 + dp[i][1]) % mod; // 为2,0,1组成;
		dp[i + 1][4] = (dp[i][4] * 2 + dp[i][1] + dp[i][2]) % mod; // 为2,0,3组成;
		dp[i + 1][5] = (dp[i][5] * 2 + dp[i][3] + dp[i][4]) % mod; // 为2,0,1,3组成;
	}
	cout << dp[n][5] << endl;
}