动态规划
一维i表示第i个球
二维表示连续几个球颜色一样,连续一个球颜色相同或两个球颜色相同
状态转移:f[i][1]说明第i个球的颜色和第i-1个球颜色不同,那么可以f[i-1][1]表示第i-1个球颜色和第i-2个球颜色不同,这样三个球颜色都不同,可行
f[i-1][2]表示第i-1个球颜色和第i-2个球颜色相同,但是因为第i个球颜色和第i-1个球颜色不同,所以可行
f[i][2]表示第i个球颜色和第i-1个球颜色相同,那么可以f[i-1][1]表示第i-1个球颜色和第i-2个球颜色不同,这样不会连续三个球颜色相同,所以可行
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 1000010,mod=1e9+7;
int f[N][3];
int main()
{
int n;
cin >> n;
f[1][1] = 2;
for (int i = 2; i <= n; i++) {
f[i][1] = (f[i - 1][1]%mod + f[i - 1][2]%mod)%mod;
f[i][2] = f[i - 1][1]%mod;
}
cout << (f[n][1] + f[n][2])%mod << endl;
return 0;
}