这个题最开始用了错误的想法,后来用BFS写的,果断超时,结果看到用的STL的queue,然后就改为数组,一交就AC了。思路那些都比较简单。
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
const int inf=1<<29;
const int maxn=1<<20;
struct Node
{
int s;
int ans;
Node(){}
Node(int ss,int sans):s(ss),ans(sans){}
}q[maxn];
bool vis[maxn];
int s,a[30];
bool check(int val)
{
int ss=s;
for(int i=0;i<20;i++)
if((val>>i)&1)
{
ss^=1<<i;
if(i>0)
ss^=1<<(i-1);
if(i<19)
ss^=1<<(i+1);
}
if(ss==0)
return true;
return false;
}
int BFS()
{
memset(vis,0,sizeof(vis));
int pre=0,last=1;
q[last++]=(Node(0,0));
vis[0]=1;
while(pre<last)
{
Node u=q[pre++];
for(int i=0;i<20;i++)
{
if(!vis[u.s|(1<<i)])
{
if(check(u.s|(1<<i)))
return u.ans+1;
vis[u.s|(1<<i)]=1;
q[last++]=Node(u.s|(1<<i),u.ans+1);
}
}
}
}
int main()
{
while(scanf("%d",&a[0])!=EOF)
{
if(a[0])
s=1;
else
s=0;
for(int i=1;i<20;i++)
{
scanf("%d",&a[i]);
if(a[i])
s|=(1<<i);
}
if(s==0)
{
printf("0\n");
continue;
}
printf("%d\n",BFS());
}
return 0;
}