BZOJ 1088: [SCOI2005]扫雷Mine

1088: [SCOI2005]扫雷Mine

Time Limit: 10 Sec  Memory Limit: 162 MB
[Submit][Status][Discuss]

Description

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

Input

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

Output

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

Sample Input

2

1 1

Sample Output

2

HINT

 

Source

 

题目链接

确定第一列就可以确定整个的情况，答案为0或1或2，枚举一下就OK了

#include<bits/stdc++.h>

using namespace std;

template <typename tn> void read (tn & a) {
    tn x = 0, f = 1;
    char c = getchar();
    while (c < '0' || c > '9'){ if (c == '-') f = -1; c = getchar(); }
    while (c >= '0' && c <= '9'){ x = x * 10 + c - '0'; c = getchar(); }
    a = f == 1 ? x : -x;
}

const int MAXN = 11000;
int n;
int a[MAXN];
bool f[MAXN];

bool check() {
    for (int i = 2; i <= n; ++i) {
        int x = a[i - 1] - f[i - 2] - f[i - 1];
        if (x < 0 || x > 1) return 0;
        f[i] = x;
    }
    if (a[n] == f[n] + f[n - 1]) return 1; else return 0;
}

int main() {
    read(n);
    for (int i = 1; i <= n; ++i) {
        read(a[i]);
    }
    int ans = 0;
    f[1] = 0;
    ans += check();    
    f[1] = 1;
    ans += check();
    cout << ans << "\n";
    return 0;
}