题意:
在一块2*N的话画布上有三种不同的积木,问恰好拼完 N 列的时候有多少不同的方案?
思路:
一看是 2*N 的画布,且数据量很大,所以不用多想,咱们已经锁定dp了。dp就想好状态表示和转移方程。
状态表示:
我们用 f [N][3] 这个数组来表示不同状态下的方案数,f [i][0] 表示当前为第 i 列,且前i-1列已经拼满,且第 i 列也拼满的方案数;f [i][1] 表示当前为第 i 列,且前 i-1 列已经拼满,且第 i 列差一个拼满,由于第 i 列有两个格子,那么差一个格子拼满是哪个格子呢?所以我们增加一个 f [i][2],与 f [i][1]来表示差一个拼满的状态。即 f [i][1]表示第 i 列差下面一个格子拼满, f [i][2]表示第 i 列差上面一个格子拼满。
状态转移:
f [i][0] 第i列刚好拼满有这么几种情况 :
第 i 列刚好拼满可能由这几种方式结尾,下面的方程就是以当前方式结尾的转移方程。
注意有个 f [i-2][0], 当i < 2 时,我们可以给只有一列的情况自己特判赋值
i>=2时,f [i][0]=(f [i-1][0]+f [i-1][1]+f [i-1][2]+f [i-2][0])%mod;
f [i][1] 第i列差下面一个格子拼满有这么几种情况:
同理 因为有 f [i-2]的存在, i < 2 我们特判赋值
i >= 2时 , f [i][1]=(f [i-1][2]+f [i-2][0])%mod;
f [i][2] 第i列差上面一个格子拼满有这么几种情况:
同理 因为有 f [i-2]的存在, i < 2 我们特判赋值
i >= 2时 , f [i][2]=(f [i-1][1]+f [i-2][0])%mod;
这样我们就愉快地可以写代码啦~ 状态转移想好后码起来非常快~
代码:
#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e7+10;
int n;
const int mod=1000000007;
long long f[MAXN][3];
int main()
{
cin>>n;
f[0][0]=1;
f[1][0]=1;
f[2][0]=2;
f[2][1]=1;
f[2][2]=1;
for(int i=3;i<=n;i++)
{
f[i][0]=(f[i-1][0]+f[i-1][1]+f[i-1][2]+f[i-2][0])%mod;
f[i][1]=(f[i-1][2]+f[i-2][0])%mod;
f[i][2]=(f[i-1][1]+f[i-2][0])%mod;
}
cout<<f[n][0]%mod;
return 0;
}