Vijos-P1193 扫雷

题意：

相信大家都玩过扫雷的游戏。那是在一个n*n的矩阵里面有一些雷，要你根据一些信息找出雷来。万圣节到了，“余”任过流行起了一种简单的扫雷游戏，这个游戏规则和扫雷一样，如果某个格子没有雷，那么它里面的数字表示和他8连通的格子里面雷的数目。现在棋盘是n*2的，第一列里某些格子是雷，而第二列没有雷，如：
o 1
* 2
* 3
* 2
o 2
* 2
* 2 ('*'代表有雷，'o'代表无雷)
由于第一类的雷有可能有多种方案满足第二列的数的限制，你的任务即根据第二列的信息求第一列雷有多少中摆放方案。

格式

输入格式

第一行为N，第二行有N个数，依次为第二列的格子中的数。（1<=N<=10000）

输出格式

一个数，即第一列中雷的摆放方案数。

样例1

样例输入1[复制]

2
1 1

样例输出1[复制]

2

限制

1s

思路：用f[i][j]表示第i行状态为j的个数，因为当前行只关系它旁边的三个格子，所以我们用二进制表示，最大为111，确定当前行的时候就可以得到前一行的后两个格子的状态，注意第一行和最后一行的特殊处理

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

int sl[8] = {0,1,1,2,1,2,2,3};

int n,a[10010];
int f[10010][8];

int main(){
    scanf("%d",&n);
    for (int i = 1; i <= n; i++)
        scanf("%d",&a[i]);
    f[0][0] = 1;
    f[0][1] = 1;
    for (int i = 1; i < n; i++)
        for (int j = 0; j < 8; j++)
            if (sl[j] == a[i])
                for (int k = 0; k <= 1; k++)
                    f[i][j] += f[i-1][(k<<2)+(j>>1)];
    int sum = 0;
    for (int i = 0; i < 8; i++)
        if (sl[i] == a[n] && i == (i>>1<<1))
            for (int k = 0; k <= 1; k++)
                sum += f[n-1][(k<<2)+(i>>1)];
    printf("%d\n",sum);
    return 0;
}