洛谷 P2327 [SCOI2005]扫雷

P2327 [SCOI2005]扫雷

题目描述

输入输出格式

输入格式：

 

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

 

输出格式：

 

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

 

输入输出样例

输入样例#1：

2
1  1

输出样例#1：

2

#include<iostream>
#include<cstdio>
using namespace std;
int n,a[10001];
int ans,b[10001];
int check(){
    b[0]=0;
    for(int i=1;i<=n;i++){
        b[i+1]=a[i]-b[i-1]-b[i];
        if(b[i]>1||b[i]<0) return 0;
    }
    return b[n+1]==0; 
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    b[1]=1;
    ans+=check();
    b[1]=0;
    ans+=check();
    printf("%d",ans);
}