【bzoj1088】扫雷

见过水的，没见过这么水的

Description

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

Input

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

Output

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

Sample Input

2
1 1

Sample Output

2

 

Solution

 

题解，枚举第一个是不是雷

显然的，后面的每一个是不是雷就显然了

 

#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#define il inline
using namespace std;
int n,a[20001],b[20001];
il bool chk(){
    for(int i=1;i<=n;i++)
        if(a[i]>3) return false;
    if(a[1]>2||a[n]>2) return false;
    return true;
}
il int work(int k){
    b[1]=k;
    for(int i=1;i<=n;i++){
        b[i+1]=a[i]-b[i-1]-b[i];
    }
    b[0]=0;b[n+1]=0;
    for(int i=1;i<=n;i++)
        if(b[i]<0||b[i]>1) return 0;
    for(int i=1;i<=n;i++){
        if(b[i-1]+b[i]+b[i+1]!=a[i]) return 0;
    }
    return 1;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    if(!chk()){
        cout<<"0";
        return 0;
    }
    cout<<work(1)+work(0)<<endl;
    return 0;
}