[SCOI2005]扫雷

题目描述

输入输出格式

输入格式：

 

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

 

输出格式：

 

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

 

输入输出样例

输入样例#1：

2
1  1

输出样例#1：

2

这题神套路..答案只有三种0,1,2

因为只要第一个的状态确定了之后,后面的要么无解,要么只有一种解.

因为每个点要么有雷要么无雷.

所以就可以先枚举第一个点是否有雷,然后DP判断是否有解.

每个点的雷数f[i]=a[i-1]-f[i-1]-f[i-2].

若算出来不是0或1就无解了.

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<complex>
#include<queue>
#include<stack>
#include<cmath>
#include<set>
#include<vector>
#define maxn 10010
using namespace std;
int f[maxn],a[maxn],n;
inline bool check(){
  for(int i=2;i<=n;i++){
    f[i]=a[i-1]-f[i-1]-f[i-2];
    if(f[i]!=0 && f[i]!=1) return 0;
  }
  if(f[n]+f[n-1]!=a[n]) return 0;
  return 1;
}
int main(){
  int ans=0;
  scanf("%d",&n);
  for(int i=1;i<=n;i++)
    scanf("%d",&a[i]);
  f[1]=0;
  if(check())ans++;
  memset(f,0,sizeof(f));
  f[1]=1;
  if(check())ans++;
  printf("%d",ans);
  return 0;
}