【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[],b[];
il bool chk(){
    for(int i=;i<=n;i++)
        if(a[i]>) return false;
    if(a[]>||a[n]>) return false;
    return true;
}
il int work(int k){
    b[]=k;
    for(int i=;i<=n;i++){
        b[i+]=a[i]-b[i-]-b[i];
    }
    b[]=;b[n+]=;
    for(int i=;i<=n;i++)
        if(b[i]<||b[i]>) return ;
    for(int i=;i<=n;i++){
        if(b[i-]+b[i]+b[i+]!=a[i]) return ;
    }
    return ;
}
int main(){
    scanf("%d",&n);
    for(int i=;i<=n;i++){
        scanf("%d",&a[i]);
    }
    if(!chk()){
        cout<<"";
        return ;
    }
    cout<<work()+work()<<endl;
    return ;
}