【dfs搜索】NOIP2011day3 Mayan游戏

据说这是noip中最难的搜索题……

很明显是一个搜索。然而我竟以为是ID搜索，无限TLE %>_<%

至于实现起来比较难的部分就是掉落和消除了。这两个函数会调用很多次，写丑了会无限TLE的…

消除函数：

bool remove(int temp[][10])
{
bool pos[10][10]={0} ,flag=0;
for(int i=1;i<=5;++i)
for(int j=1;j<=7;++j)
if(temp[i][j])
{
if(i<=3&&temp[i][j]==temp[i+1][j]&&temp[i+1][j]==temp[i+2][j])
pos[i][j]=pos[i+1][j]=pos[i+2][j]=1;
if(j<=5&&temp[i][j]==temp[i][j+1]&&temp[i][j+1]==temp[i][j+2])
pos[i][j]=pos[i][j+1]=pos[i][j+2]=1;
}
for(int i=1;i<=5;++i)
for(int j=1;j<=7;++j)
if(pos[i][j])
temp[i][j]=0 ,flag=1 ;
return flag;
}

掉落函数：

void falldown(int temp[][10])
{
int k ,tmp ;
for(int i=1;i<=5;++i)
{
k=0;
for(int j=1;j<=7;++j)
{
tmp=temp[i][j] ,temp[i][j]=0 ;
if(tmp)temp[i][++k]=tmp;
}
}
}

还有一个很重要的想法：向左可以有向右替代（除了有一边是0）。所以要特判这种情况。

还有一个弱的剪枝：当图中有一种颜色的个数少于3时，肯定不能消除完。

付代码

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

int cnt[15] ,color[10][10] ,n ,lin[6][3] ;

bool remove(int temp[][10])
{
bool pos[10][10]={0} ,flag=0;
for(int i=1;i<=5;++i)
for(int j=1;j<=7;++j)
if(temp[i][j])
{
if(i<=3&&temp[i][j]==temp[i+1][j]&&temp[i+1][j]==temp[i+2][j])
pos[i][j]=pos[i+1][j]=pos[i+2][j]=1;
if(j<=5&&temp[i][j]==temp[i][j+1]&&temp[i][j+1]==temp[i][j+2])
pos[i][j]=pos[i][j+1]=pos[i][j+2]=1;
}
for(int i=1;i<=5;++i)
for(int j=1;j<=7;++j)
if(pos[i][j])
temp[i][j]=0 ,flag=1 ;
return flag;
}

void falldown(int temp[][10])
{
int k ,tmp ;
for(int i=1;i<=5;++i)
{
k=0;
for(int j=1;j<=7;++j)
{
tmp=temp[i][j] ,temp[i][j]=0 ;
if(tmp)temp[i][++k]=tmp;
}
}
}

bool check(int temp[][10])
{
for(int i=1;i<=5;++i)
for(int j=1;j<=7;++j)
if(temp[i][j])return 0;
return 1;
}

void dfs(int code,int temp[][10])
{
if(code>n)
{
if(check(temp))
{
for(int i=1;i<=n;++i)
{
if(lin[i][2])printf("%d %d -1\n",lin[i][0],lin[i][1]-1);
else printf("%d %d 1\n",lin[i][0]-1,lin[i][1]-1);
}
exit(0);
}
return;
}

int tmp[10][10]={0} ;
memset(cnt,0,sizeof cnt);
for(int i=1;i<=5;++i)
for(int j=1;j<=7;++j)
++cnt[temp[i][j]];
for(int i=1;i<=10;++i)
if(cnt[i]==1||cnt[i]==2)
return;
for(int i=1;i<5;++i)
for(int j=1;j<=7;++j)
if(temp[i][j]!=temp[i+1][j])
{
memcpy(tmp,temp,sizeof tmp);
lin[code][0]=i ,lin[code][1]=j ,lin[code][2]=!temp[i][j];
swap(tmp[i][j],tmp[i+1][j]);
falldown(tmp);
while(remove(tmp))
falldown(tmp);
dfs(code+1,tmp);
}
}

int main()
{
scanf("%d",&n);
for(int i=1;i<=5;++i)
for(int j=1;;++j)
{
scanf("%d",&color[i][j]);
if(!color[i][j])break;
}
dfs(1,color);
puts("-1");
return 0;
}