【题解】
设给出的信息为$a[i]$,每个格子的雷数为$f[i]$
我们可以发现$f[i]=a[i-1]-f[i-1]-f[i-2]$
也就是说只要确定了前两个格子的情况,全部的情况都可以计算了。
那么我们枚举前两个格子的情况,检查是否合法即可。
#include#include #define N 200010#define rg registerusing namespace std;int n,m,a[N],f[N],ans; inline int read(){ int k=0,f=1; char c=getchar(); while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar(); while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar(); return k*f;}inline int check(){ for(rg int i=3;i<=n;i++){ f[i]=a[i-1]-f[i-2]-f[i-1]; if(f[i]<0) return 0; } if(f[n-1]+f[n]!=a[n]) return 0; return 1;}int main(){ n=read(); for(int i=1;i<=n;i++) a[i]=read(); if(a[1]==0){ f[1]=f[2]=0; ans+=check(); } if(a[1]==1){ f[1]=0; f[2]=1; ans+=check(); f[1]=1; f[2]=0; ans+=check(); } if(a[1]==2){ f[1]=f[2]=1; ans+=check(); } return printf("%d\n",ans),0;}