【NOIP模拟赛】盟主的忧虑

在这里插入图片描述

题解:

显然任意所求路径就只经过一条密道,然后呢,画图发现就是一个个环,然后可以用并查集来维护属于一个环或者是直接赋值

#include<bits/stdc++.h>
using namespace std;
long long read() {
	long long num=0;
	char ch=getchar();
	while(ch>'9'||ch<'0') {
		ch=getchar();
	}
	while(ch>='0'&&ch<='9') {
		num=(num<<1)+(num<<3)+ch-'0';
		ch=getchar();
	}
	return num;
}
struct oo{
	int next;
	int ver;
	int id;
}edge[200005];
struct ooo{
	int s;
	int e;
	int val;
}md[200005];
int head[100005],tot,d[100005],fa[100005],up[100005],in[100005],ans[100005];
void dfs(int now){
	for(int i=head[now];i;i=edge[i].next){
		int y=edge[i].ver;
		if(d[y])continue;
		d[y]=d[now]+1;
		in[y]=edge[i].id;
		fa[y]=now;
		dfs(y);
	}
}
void add(int a,int b,int c){
	edge[++tot].next=head[a];
	edge[tot].ver=b;
	edge[tot].id=c;
	head[a]=tot;
}
bool cmp(ooo a,ooo b){
	return a.val<b.val;
}
int finds(int x){
	if(up[x]!=x)
		up[x]=finds(up[x]);
	return up[x];
}
int main(){
	int n,m,a,b;
	n=read();
	m=read();
	for(int i=1;i<n;i++){
		a=read(),b=read();
		add(a,b,i);
		add(b,a,i);
	}
	d[1]=1;
	dfs(1);
	for(int i=1;i<=m;i++){
		md[i].s=read();
		md[i].e=read();
		md[i].val=read();
	}
	sort(md+1,md+m+1,cmp);
	for(int i=1;i<=n;i++)
		up[i]=i;
	for(int i=1;i<=m;i++){
		int u=finds(md[i].s),v=finds(md[i].e);
		while(u!=v){
			if(d[u]<d[v])swap(u,v);
			ans[in[u]]=md[i].val;
			u=up[finds(u)]=finds(fa[u]);
		}
	}
	for(int i=1;i<n;i++){
		if(ans[i])
			printf("%d\n",ans[i]);
		else
			puts("-1");
	}
	return 0;
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值