Kruskal写挂了求助
查看原帖
Kruskal写挂了求助
263414
Sktic楼主2022/9/10 12:55

改了亿些错误现在代码是这样

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
const int INF=0x3f3f3f3f;
typedef long long ll;
struct edge
{
	int v,w;
};
struct edd
{
	int u,v,w;
}bc[maxn];
bool cmp(edd x,edd y)
{
	return x.w>y.w;
}
vector<edge>c[maxn];
int hen[maxn],lg[maxn],deep[maxn],vis[maxn];
int fa[maxn][30],mw[maxn][30];
void init(int n)
{
	for(int i=1;i<=n;i++)
		lg[i]=lg[i-1]+(1<<lg[i-1]==i);
	return;
}
void dfs1(int x)
{
	vis[x]=1;
	for(int i=0;i<c[x].size();i++)
	{
		int k=c[x][i].v;
		if(vis[k])
			continue;
		deep[k]=deep[x]+1;
		fa[k][0]=x;
		mw[k][0]=c[x][i].w;
		dfs1(k);
	}
	return;
}
void dfs2(int n)
{
	for(int i=1;i<=30;i++)
		for(int j=1;j<=n;j++)
			fa[j][i]=fa[fa[j][i-1]][i-1],mw[j][i]=min(mw[j][i-1],mw[fa[j][i-1]][i-1]);
	return;
}
int find(int x)
{
	if(hen[x]!=x)
		return hen[x]=find(hen[x]);
	return hen[x];
//	return (x==hen[x]?hen[x]:hen[x]=find(hen[x]));
}
void kruskal(int n,int m)
{
	sort(bc+1,bc+m+1,cmp);
	for(int i=1;i<=n;i++)
		hen[i]=i;
	for(int i=1;i<=m;i++)
	{
		int u=bc[i].u,v=bc[i].v,w=bc[i].w;
		if(find(u)!=find(v))
		{
			c[u].push_back(edge{v,w});
			c[v].push_back(edge{u,w});
			hen[find(u)]=find(v);
		}
	}
	return;
}
int lca(int x,int y)
{
	if(find(x)!=find(y))
		return -1;
	int ans=INF;
	if(deep[x]<deep[y])
		swap(x,y);
	for(int i=20;i>=0;i--)
		if(deep[fa[x][i]]>=deep[y])
		ans=min(ans,mw[x][i]),x=fa[x][i];
	if(x==y)
		return ans;
	for(int i=20;i>=0;i--)
		if(fa[x][i]!=fa[y][i])
			ans=min(ans,min(mw[x][i],mw[y][i])),x=fa[x][i],y=fa[y][i];
	ans=min(ans,min(mw[x][0],mw[y][0]));
	return ans;
}
int main()
{
//	ios::sync_with_stdio(false);
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		int u,v,w;
		cin>>u>>v>>w;
		bc[i]=edd{u,v,w};
	}
	kruskal(n,m);
	init(n);
	for(int i=1;i<=n;i++)
	{
		if(!vis[i])
		{
			deep[i]=0;
			dfs1(i);
			fa[i][0]=i;
			mw[i][0]=INF;
		}
	}
	dfs2(n);
	int q;
	cin>>q;
	for(int i=1;i<=q;i++)
	{
		int x,y;
		cin>>x>>y;
		cout<<lca(x,y)<<endl;
	}
	return 0;
}
/*
5 7
4 3 4440
3 1 22348
1 3 28368
2 4 25086
5 3 6991
4 3 10638
3 1 11106
4
4 5
1 3
5 4
2 
*/ 
2022/9/10 12:55
加载中...