【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;
}