递归小练习1

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

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

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

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

解题：
用递归的思路想,子问题是什么，这题的子问题就是当n减少时的多种情况然后递归的结束点是在n=1或2的哪几种情况
所以我用一个flag来代表相同n时不同的方块情况

#include<iostream>
#include<queue>
#include<stack>
#include<algorithm>
#include<map>

using namespace std;
map<int,int> mp;
int ans(int n,int flag){
//5个结束条件
    if(n==1&&flag==0) return 1;
    if(n==2&&flag==1) return 1;
    if(n==2&&flag==2) return 1;
    if(n==2&&flag==3) return 1;
    if(n==2&&flag==4) return 1;
//5个情况的递归情况
    if(flag==0){
    if(mp.count(n)==1) return mp[n];
    else {
    mp[n]=ans(n-1,1)+ans(n-1,2)+ans(n-1,3)+ans(n-1,4)+ans(n-1,0);
    return mp[n];
    }
    }
    if(flag==1)
    return ans(n-1,0)+ans(n-1,1);
    if(flag==2)
    return ans(n-1,0)+ans(n-1,2);
    if(flag==3)
    return ans(n-2,0);
    if(flag==4)
    return ans(n-2,0);
}
int main(){
   int n;
   cin>>n;
   cout<<ans(n,0)%10000<<endl;
}