40分 并查集写的 RE了好多点 求助
查看原帖
40分 并查集写的 RE了好多点 求助
170047
小渣青999楼主2020/8/23 21:45
#include<iostream>
#include<cstdio>
#include<algorithm> 
#include<queue>
#include<vector>
using namespace std;
int a[1000010],fa[1000010];
int findfa(int x)
{
	if(fa[x]==x) return x;
	else return fa[x]=findfa(fa[x]);
}
void mg(int x,int y)
{
	int fx=findfa(x),fy=findfa(y);
	if(fx!=fy) fa[fx]=fy;
}

struct edge
{
	int u,v,cost;
}e[1000010];

bool cmp(edge a,edge b)
{
	return a.cost<b.cost;
}
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=0;i<n;i++)
	    for(int j=0;j<m;j++)
	    {
	    	scanf("%d",&a[i*m+j]);
	    	fa[i*m+j]=i*m+j;
		}
	int k=-1;
	for(int i=0;i<n-1;i++)
	    for(int j=0;j<m-1;j++)
	    {
	    	e[++k].u=(i*m+j);
	    	e[k].v=(i*m+j+1);
	    	e[k].cost=max(a[e[k].u],a[e[k].v]);
	    	
	    	e[++k].u=(i*m+j);
	    	e[k].v=((i+1)*m+j);
	    	e[k].cost=max(a[e[k].u],a[e[k].v]);
		}
	for(int i=1;i<m;i++)
	{
		mg(0,i);mg((n-1)*m,(n-1)*m+i);
	}
	
	sort(e,e+k+1,cmp);
	/*for(int i=0;i<k+1;i++) 
	 printf("%d %d cost:%d\n",e[i].u,e[i].v,e[i].cost);
	 */
	 long long int ans;
	for(int i=0;;i++)
	{
		int fs=findfa(0),fe=findfa((n-1)*m);
		if(fs==fe) break;
		int u=e[i].u,v=e[i].v;
		int fu=findfa(u),fv=findfa(v);
		if(fu==fv) continue;
		//printf("%d %d cost:%d\n",u,v,e[i].cost);
		ans=e[i].cost;
		fa[fu]=fv;
	}
	printf("%lld",ans);
	return 0;
}

谢谢好心的大佬

2020/8/23 21:45
加载中...