P8784积木画

P8784积木画

题目

题目链接

分析

每一步有三种状态：
固定长度是列=2
0. 第一行积木比第二行积木多一个

  状态1在第一行添加一个I型积木
  状态2添加一个L型积木

1. 第二行积木比第一行积木多一个

  状态0在第二行添加一个I型积木
  状态2添加一个L型积木

2. 两行相等。

  状态0添加一个L型积木
  状态1添加一个L型积木
  状态2添加两个L型积木
  状态2添加一个纵向I型积木
  状态2添加两个横向I型积木

dp[i][j]意义：在第i列，状态为j时，积木摆放方案数
初始化：初始化的状态一定为2，方案数为1（dp[0][2]=1）
转移方程：见代码。

代码

#include<bits/stdc++.h>
using namespace std;

const int mod = 1e9+7;
int n,dp[10000005][3];

int main(){
	cin>>n;
	dp[0][2]=1;
	for(int i = 1;i<=n;i++){
		dp[i][0]=(dp[i-1][1]+dp[i-2][2])%mod;
		dp[i][1]=(dp[i-1][0]+dp[i-2][2])%mod;
		dp[i][2]=((dp[i-1][0]+dp[i-1][1])%mod+(dp[i-1][2]+dp[i-2][2])%mod)%mod;
		
	}
	cout<<dp[n][2];
	return 0;
}

注意：不要设为long long，会超时。