洛谷 P1990 覆盖墙壁

P1990 覆盖墙壁

题目描述

你有一个长为N宽为2的墙壁，给你两种砖头：一个长2宽1，另一个是L型覆盖3个单元的砖头。如下图：

0 0 0 00 砖头可以旋转，两种砖头可以无限制提供。你的任务是计算用这两种来覆盖N*2的墙壁的覆盖方法。例如一个2*3的墙可以有5种覆盖方法，如下：

012 002 011 001 011

012 112 022 011 001

注意可以使用两种砖头混合起来覆盖，如2*4的墙可以这样覆盖：

0112 0012

给定N，要求计算2*N的墙壁的覆盖方法。由于结果很大，所以只要求输出最后4位。例如2*13的覆盖方法为13465,只需输出3465即可。如果答案少于4位，就直接输出就可以，不用加0，如N=3，时输出5。

输入输出格式

输入格式：

 

一个整数N（1<=N<=1000000），表示墙壁的长。

 

输出格式：

 

输出覆盖方法的最后4位，如果不足4位就输出整个答案。

 

输入输出样例

输入样例#1： 复制

13

输出样例#1： 复制

3465
思路：dp。注意分情况，不要漏解。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n;
int f[1000010][2];
int main(){
    scanf("%d",&n);
    f[0][0]=1;f[1][0]=1;
    for(int i=2;i<=n;i++){
        f[i][0]=(f[i-1][1]+f[i-1][0]+f[i-2][0])%10000;
        f[i][1]=(f[i-1][1]+2*f[i-2][0])%10000;
    }
    cout<<f[n][0];
}