51nod 1416两点

题目描述

给你一个$n\ast m（1\le n,m\le 50）$的矩阵，用不同的大写字母来表示不同的颜色，找出一个只包含一种颜色的环。

例如这张图片中的4个蓝点就构成了一个环。

解题思路

dfs思路

对于这种要往四周找的题，我们可以用dfs做，从某个点开始向四周dfs，不能后退，如果能够再次顺着相同颜色回到这个点，那么就表示有能走回来的路，也就是有一个环。

dfs代码

#include<bits/stdc++.h>
using namespace std;
char c[55][55];
int n,m,fx,fy; 
const int xy[4][2]={{1,0},{0,1},{-1,0},{0,-1}};
bool v[55][55];
bool pd(int x,int y)
{
	return x>0&&x<=n&&y>0&&y<=m;
}
bool dfs(int x,int y,char s,int k)//s为当前颜色，k为当前长度
{
	if(x==fx&&y==fy&&k>=4) return 1;//如果回到了这个点就有环
	for(int i=0;i<4;i++)
	{
		int xx=x+xy[i][0],yy=y+xy[i][1];
		if(pd(xx,yy)&&c[xx][yy]==s&&!v[xx][yy])
		{
			if(xx==fx&&yy==fy&&k>=4) return 1;//如果到了原点
			v[xx][yy]=1;//标记走过
			if(dfs(xx,yy,s,k+1)) return 1;
			v[xx][yy]=0;
		}
	}
	return 0;
}
int main()
{
	while(cin>>n>>m)
	{
		bool f=true;
		memset(v,0,sizeof(v));
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=m;j++)
				cin>>c[i][j];
		}
		for(int i=1;i<=n;i++)
			for(int j=1;j<=m;j++)
			{
				fx=i,fy=j;
				if(dfs(i,j,c[i][j],1)){//每一个点都试一次
					f=0;
				}
			}
		if(f) cout<<"No"<<endl;
		else cout<<"Yes"<<endl;
		break;
	}
	return 0;
}

并查集思路

对于颜色相同且相邻的两个点$x$和$y$，如果它们不属于同一集合，就把它们加进一个集合；如果它们属于同一集合，那么就表示有几个点将它们连成了一个集合，那么就存在了一个环。

并查集代码

#include<bits/stdc++.h>
using namespace std;
int f[25500],n,m;
char c[55][55];
int find(int x)
{
   if(f[x]==x) return x;
   else return f[x]=find(f[x]);
}
void join(int x,int y)
{
   f[find(x)]=f[find(y)];
}
int main()
{
   cin>>n>>m;
   for(int i=1;i<=n;i++)
   	for(int j=1;j<=m;j++)
   	{
   		cin>>c[i][j];
   		f[(i-1)*m+j]=(i-1)*m+j;//将矩阵转化为线性结构
   	}
   bool flag=0;
   for(int i=1;i<=n;i++)
   {
   	for(int j=1;j<=m;j++)
   	{
   		if(c[i][j]==c[i-1][j])//如果是在左右两边
   		{
   			if(find((i-2)*m+j)==find((i-1)*m+j)) flag=1;//如果这两个是同一集合
   			else join((i-2)*m+j,(i-1)*m+j);//否则加进一个集合
   		} 
   		if(c[i][j]==c[i][j-1])//如果是上下
   		{
   			if(find((i-1)*m+j)==find((i-1)*m+j-1)) flag=1;
   			else join((i-1)*m+j,(i-1)*m+j-1);
   		}
   	}
   }
   if(flag) cout<<"Yes";else cout<<"No";
   return 0;
}