hiho刷题日记——第三十天小Hi小Ho的惊天大作战：扫雷·一

题目

简化版扫雷，只有一行。

给出一行的周围3格中是雷的数量。

判断哪些一定是雷，一定不是雷。

思路

因为给出的信息是环环相扣的，不用一个一个去试。

只需要试第一格是否为雷就可以通过数据算出后面是否为雷。

所以只用试两次，

第一次把一格设为雷，算出后面的情况；第二次把第一格设为不是雷，算出后面的情况。

若两次都算出来了，则将两次的情况对比，没有变化的部分就是答案。

若只有一次满足条件，则那一次直接就是答案。

代码

#include<cstdio>
#include<cstring>
using namespace std;

const int MAXN=100000+2;
int N,M,a[MAXN],ans[2][MAXN],count[]={0,0},stat1[MAXN],stat2[MAXN];
int t1,t2;

bool run(int flag,int *s)
{
s[0]=0;
s[1]=flag;
for(int i=1;i<N;i++)
{
s[i+1]=a[i]-s[i]-s[i-1];
if(s[i+1]!=0 && s[i+1]!=1) return false;
}
if(a
==s[N-1]+s
) return true;
return false;
}

int main()
{
scanf("%d",&M);
a[0]=0;
while(M--)
{
t1=t2=count[0]=count[1]=0;

scanf("%d",&N);
for(int i=1;i<=N;i++) scanf("%d",&a[i]);

if(run(0,stat1)) t1=1;
if(run(1,stat2)) t2=1;
if(t1)
{
if(t2)
{
for(int i=1;i<=N;i++)
if(stat1[i]==stat2[i]) ans[stat1[i]][count[stat1[i]]++]=i;
}
else
{
for(int i=1;i<=N;i++)
ans[stat1[i]][count[stat1[i]]++]=i;
}
}
else if(t2)
{
for(int i=1;i<=N;i++)
ans[stat2[i]][count[stat2[i]]++]=i;
}

printf("%d",count[1]);
if(count[1])
{
for(int i=0;i<count[1];i++) printf(" %d",ans[1][i]);
}
printf("\n");
printf("%d",count[0]);
if(count[0])
{
for(int i=0;i<count[0];i++) printf(" %d",ans[0][i]);
}
printf("\n");
}
return 0;
}