bzoj 1088: [SCOI2005]扫雷Mine （枚举+判定）

题目描述

传送门

题解

其实第一个位置和第二个位置是否有雷确定了，那么填的方案就确定了。
所以只需要枚举第一位置和第二个位置是否有雷即可。

代码

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdio>
#define N 20003
using namespace std;
int n,a[N];
int check(int c1,int c2)
{
    int c3=0;
    for (int i=2;i<=n-1;i++){
        c3=a[i]-c1-c2;
        if (c3<0) return 0;
        c1=c2; c2=c3;
    }
    if (c1+c2!=a[n]) return 0;
    return 1;
}
int main()
{
    freopen("a.in","r",stdin);
    scanf("%d",&n); int ans=0;
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    for (int i=0;i<=1;i++)
     for (int j=0;j<=1;j++)
      if (i+j==a[1])
       ans+=check(i,j);
    printf("%d\n",ans);
}