oj 深度优先搜索练习之神奇的矩环

深度优先搜索练习之神奇的矩环

Description
小鑫的女朋友被魔王抢走了！
魔王留给小鑫一张n*m大的表，上面有各种各样的颜色，用A-Z这26个字母来表示。魔王留给他一个任务，如果小鑫可以在这张表中找出任意一个长度大于1的环，并且这个环的颜色是相同的，魔王就把小鑫的女朋友还给他。为了从魔王手中夺回他的女朋友，小鑫请你帮忙，你能帮帮他吗？
Input
多组输入。
每组的第一行有两个整数n,m。代表表的大小。
接下来是由A-Z的一些字母所构成的n行m列的表。
1<=n,m<=200
Output
如果可以救回他的女朋友，输出Yes，否则输出No
Sample
Input
4 7
ABCBBAA
BCBCBCB
AABBCCA
ACCCBBB
10 3
AAC
ABB
BBA
AAC
CBC
CCA
CBB
CCA
CCB
BAA
Output
No
Yes
Hint

#include <cstring>
#include <iostream>
using namespace std;
char map[205][205];
int book[205][205];
int n,m;
int flag;
void dfs(int x,int y,int px,int py)
{
    book[x][y]=1;
    if(flag)//如果只有这一个点形成闭环，就删除
        return;
    if(x-1>=0 && map[x-1][y]==map[x][y])//走到没有走过的点并且颜色相同
    {
        if(book[x-1][y]!=0 && ((x-1!=px)||(y!=py)))//此点已经走过且该点的坐标不是上一点的坐标，就是已经成环
            flag=1;
        else if(!book[x-1][y])//如果该点没有走过
            dfs(x-1,y,x,y);//继续dfs
    }
    if(x+1<n && map[x+1][y]==map[x][y])
    {
        if(book[x+1][y] && ((x+1!=px) || (y!=py)))
            flag=1;
        else if(!book[x+1][y])
            dfs(x+1,y,x,y);
    }
    if(y-1>=0 && map[x][y-1]==map[x][y])
    {
        if(book[x][y-1] && ((x!=px)||(y-1!=py)))
            flag=1;
        else if(!book[x][y-1])
            dfs(x,y-1,x,y);
    }
    if(y+1<m && map[x][y+1]==map[x][y])
    {
        if(book[x][y+1] && ((x!=px)||(y+1!=py)))
            flag=1;
        else if(!book[x][y+1])
            dfs(x,y+1,x,y);
    }
}
int main()
{
    while(cin>>n>>m)
    {
        int i,j;
        flag=0;
        memset(map,0,sizeof(map));
        memset(book,0,sizeof(book));
        for(i=0;i<n;i++)
            cin>>map[i];
        for(i=0;i<n;i++)
        {
            for(j=0;j<m;j++)
            {
                if(!book[i][j])
                    dfs(i,j,i,j);
                if(flag)//剪枝优化
                    break;//剪枝优化就是避开死胡同，将不能到达的路直接删除
            }
            if(flag)//剪枝优化，如果没有形成闭环
                break;//来减少时间复杂度
        }
        if(flag)
            cout<<"Yes"<<endl;
        else//如果遍历到最后也没有形成闭环
            cout<<"No"<<endl;
    }
    return 0;
}