flood_it 方法记录

时间:2022-10-10 10:10:05

根据题面的描述,我们面临的问题无非是,每次将色块更新成什么颜色。又因为是从左上角开始更新,所以我的有了第一个想法。

将左上角的色块命名为“原色块”。

对于每个色块,定义4中状态:
0-
不属于原色块*,和原色块*不邻接,未没有进行任何操作;
1-
不属于原色块*,和原色块*邻接,尚不知道会不会被同化(因为还不知道下一步染什么色);
2-
属于原色块*,和非原色块*邻接,是上一批被同化的色块;
3-
属于原色块*,和非原色块*不邻接**;

我们可以形象地依次把这四种状态称作“未加入”,“待审核”,“新成员”,“老成员”
以下图为例(颜色是色块,数字是状态):
flood_it 方法记录

第一步比较特殊,因为需要求出最初的新队员。(之后就只需要处理新成员和待审核就可以了,因为新成员处于邻接位置,需要承担审核工作。而老成员和未加入不需要做任何工作)

首先原色块自身是新成员,然后跑一遍bfs,求出所有与原色块连通且与原色块同色的色块,将它们都标记为新成员。

之后的每一步,首先统计与新成员邻接且未加入的色块,将它们标记为待审核,并统计每种颜色出现的次数。

出现次数最多的颜色就是我们即将更新的颜色。得到这个之后新成员的工作就做完了,新成员就可以标记为老成员了。

再次遍历所有待审核的色块,若颜色和即将更新的颜色相同,则标记为新成员;否则重新标记为未加入。

将所有新成员、老成员的颜色更新。

每次都需要判断是否更新完整个图。

这个算法逻辑上是行得通的,但是太慢了,因为要反复地遍历各种状态,导致大量无用的重复运算。

正解

标算:IDA*

先不谈IDA*。我们回到这道题,从模块化编程的角度出发,想想这个程序要实现什么功能。

首先我们有更新色块的操作。对于更新色块的函数,我们需要3个形参:当前色块的横坐标、纵坐标、我们即将更新的颜色。我们仍然需要一个打标记的数组:若当前位置的颜色和即将更新的颜色相同则标记为1,反之则标记为2.所以这个函数的主要功能是给需要更新颜色的色块打标记,还没有进行上色操作。

void update_flags(int i,int j,int color)//color是即将更新的颜色 
{
    if(board[i][j]!=color)
    {
        flags[i][j]=2;//和即将更新的颜色不同,打上标记2 
        return ;
    }
    flags[i][j]=1;//和即将更新的颜色相同,打上标记1 
    for(int k=0;k<4;k++)
    {
    	//dir[][]用于枚举4个方向 
        int x=i+dir[k][0];
        int y=j+dir[k][1];
        if(x<0||y<0||x>=siz||y>=siz||flags[x][y]) continue;
        update_flags(x,y,color);//递归更新尚未打标记的色块 
    }
}

然后考虑具体的上色操作。最外层循环遍历6种颜色,然后遍历全图。若当前遍历到的位置是不同颜色且正好等于要搜索的颜色,则记录存在这种颜色,并进行\(update_flags\).若全图遍历完后发现不存在这种颜色,则继续遍历下一个颜色。如果更新了棋盘,就递归搜索,成功了就返回。注意:\(flags\)数组在这个过程中可能会被改变,为了后续的使用,我们需要在操作前拷贝\(flags\)\(tmp\)数组,操作结束后再拷贝回去。

接下来再考虑使用IDA.IDA可以理解为以迭代加深DFS的搜索框架为基础,把原来简单的深度限制加强为:若当前深度+未来估计步数>深度限制,则立即从当前分支回溯

具体到本题,我们的估价函数就应该是:如果颜色数量大于剩余步数,则行不通。所以我们还需要一个计算剩余颜色数量的函数。

int color_cnt()
{
    int ret=0;
    int s=0;
    for(int i=0;i<siz;i++)
        for(int j=0;j<siz;j++)
            if(flags[i][j]!=1)
                s|=(1<<board[i][j]);
    while(s)//检查还有几种颜色 
    {
        ret+=s&1;
        s>>=1;
    }
    return ret;
}

并且,我们要在主函数中枚举每一种最大限定深度。

最终代码呈现如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=15;
int siz;
int board[N][N];
int dir[4][2]={{0,1},{1,0},{0,-1},{-1,0}};
int flags[N][N];
int max_depth;
void update_flags(int i,int j,int color)//color是即将更新的颜色 
{
    if(board[i][j]!=color)
    {
        flags[i][j]=2;//和即将更新的颜色不同,打上标记2 
        return ;
    }
    flags[i][j]=1;//和即将更新的颜色相同,打上标记1 
    for(int k=0;k<4;k++)
    {
    	//dir[][]用于枚举4个方向 
        int x=i+dir[k][0];
        int y=j+dir[k][1];
        if(x<0||y<0||x>=siz||y>=siz||flags[x][y]) continue;
        update_flags(x,y,color);//递归更新尚未打标记的色块 
    }
}
int color_cnt()
{
    int ret=0;
    int s=0;
    for(int i=0;i<siz;i++)
        for(int j=0;j<siz;j++)
            if(flags[i][j]!=1)
                s|=(1<<board[i][j]);
    while(s)//检查还有几种颜色 
    {
        ret+=s&1;
        s>>=1;
    }
    return ret;
}
int dfs(int depth)
{
	//如果颜色数量大于剩余步数,则行不通(估价函数) 
    if(color_cnt()>max_depth-depth)
        return 0;
    //如果当前深度刚好等于最大深度,说明已经找完了 
    if(depth==max_depth)
        return 1;
    int tmp[N][N];
    int color_exist;
    memcpy(tmp,flags,sizeof(flags));
    for(int color=0;color<6;color++)
    {
        color_exist=0;
        for(int i=0;i<siz;i++)
            for(int j=0;j<siz;j++)//当前位置是不同颜色且正好等于要搜索的颜色 
                if(flags[i][j]==2&&board[i][j]==color)
                {
                    color_exist=1;
                    update_flags(i,j,color);
                }
        //颜色不存在,继续搜索 
        if(!color_exist) continue;
        //如果更新了棋盘,就递归搜索,成功则返回 
        if(dfs(depth+1)) return 1;
        memcpy(flags,tmp,sizeof(flags));
    }
    return 0;
}
int main()
{
    while(scanf("%d",&siz)&&siz)
    {
        for(int i=0;i<siz;i++)
            for(int j=0;j<siz;j++)
                scanf("%d",&board[i][j]);
        //限定最大深度 
        for(max_depth=0;;max_depth++)
        {
            memset(flags,0,sizeof(flags));
            update_flags(0,0,board[0][0]);
            if(dfs(0)) break;
        }
        printf("%d\n",max_depth);
    }
    return 0;
}

参考