题目描述
传送门
题解
其实第一个位置和第二个位置是否有雷确定了,那么填的方案就确定了。
所以只需要枚举第一位置和第二个位置是否有雷即可。
代码
#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);
}