题意:有三列n行的雷区。已知第二行没有雷,并且第二行周围的雷已知,问你这样的雷区存在多少个。
题解:根据扫雷的规则,已知第一列那么可以递推出第二列,所以枚举第一列雷的个数,然后递推下一列判断一下是否满足题意即可。
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int mod=100000007;
char s[100005];int a[100005];
int p[3] = {1, 2, 1};int n;
ll solve()
{
ll ans=0;
for(int i=0;i<=2&&i<=(s[1]-'0');i++)
{
if(s[1]-'0'-i>2) continue;
a[1]=s[1]-'0'-i;
for(int j=2;j<=n;j++)
{
a[j]=(s[j-1]-'0')-a[j-1]-a[j-2];
}
if(a[n]+a[n-1]!=(s[n]-'0')) continue;
ll res=1;
for(int j=1;j<=n;j++)
{
if(a[j]==1)
res=(res*2);
if(a[j]!=1&&a[j]!=0&&a[j]!=2) res=0;
if (res > mod) res %= mod;
}
ans=(ans+res)%mod;
}
return ans;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
memset(a,0,sizeof(a));
scanf("%s",s+1);n=strlen(s+1);
int flag=1;
for(int i=1;i<=n;i++)
{
if((i==1||i==n)&&s[i]-'0'>4) {printf("0\n");flag=0;break;
}
if(s[i]-'0'>6) {printf("0\n");flag=0;break;
}
}
if(!flag) continue;
printf("%lld\n",solve());
}
return 0;
}