题目大意:
给出一棵有 N N N个节点的树的 N − 1 N-1 N−1条边,它们通过这些边不需要消耗费用,然后给出 M M M条密道,需要花费 w i w_i wi经过,将一开始的 N − 1 N-1 N−1条边分别断掉,然后每次求出两个点断掉以后通过剩下的边和密道使这两点连通花费的最少费用,无法连通就是-1。
N , M < = 100 , 000 N,M<=100,000 N,M<=100,000
分析:
容易发现,去掉一条树边之后,经过的密道至多只有一条,对于一条密道(u,v,w),在u到v的路径上的边的答案至多
为w,暴力更新即可。
将所有密道按照权值从小到大排序。
对于一条密道(u,v,w),如果u到v的路径上的边中存在边之前还没有被覆盖过,那么说明这条边的答案就是w.
可以用并查集维护,维护每个集合深度最小的节点,对于密道(u,v,w),每次u都在它所在集合中找到深度最小的
点,这个点与父亲的连边一定就是上述的边,将这条边的答案更新为w,然后将这个节点与其父亲合并,直到u所
在集合的深度最小的节点是小于u和v的lca的,对v做同样的过程即可。
代码:
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
#define N 100005
using namespace std;
struct Node { int rank, To, nxt; }e[N*2];
struct Code { int u, v, w; }a[N];
int Answer[N], belong[N], Oank[N], deep[N], fa[N], ls[N], n, m, cnt;
int read(int &x)
{
int f = 1; x = 0; char s = getchar();
while (s < '0' || s > '9') { if (s == '-') f = -1; s = getchar(); }
while (s >= '0' && s <= '9') { x = x * 10 + (s - '0'); s = getchar(); }
x = x * f;
}
void Addedge(int u, int v, int rank)
{
e[++cnt].To = v, e[cnt].nxt = ls[u], e[cnt].rank = rank, ls[u] = cnt;
e[++cnt].To = u, e[cnt].nxt = ls[v], e[cnt].rank = rank, ls[v] = cnt;
}
void dfs(int x, int father)
{
fa[x] = father;
for (int i = ls[x]; i; i = e[i].nxt)
{
if (e[i].To == father) continue;
int y = e[i].To; deep[y] = deep[x] + 1;
dfs(y, x); Oank[y] = e[i].rank;
}
}
int Find(int x)
{
if (belong[x] == x) return x;
return (belong[x] = Find(belong[x]));
}
bool cmp(Code aa, Code bb)
{
return aa.w < bb.w;
}
int main()
{
freopen("worry.in", "r", stdin);
freopen("worry.out", "w", stdout);
read(n); read(m);
for (int i = 1; i < n; i++)
{
int u, v; read(u); read(v); Addedge(u, v, i);
}
for (int i = 1; i <= n; i++) belong[i] = i;
deep[1] = 1, dfs(1, 0);
for (int i = 1; i <= m; i++)
read(a[i].u), read(a[i].v), read(a[i].w);
sort(a + 1, a + m + 1, cmp);
for (int i = 1; i <= m; i++)
{
int xx = Find(a[i].u), yy = Find(a[i].v);
while (xx != yy)
{
if (deep[xx] < deep[yy]) swap(xx, yy);
Answer[Oank[xx]] = a[i].w, belong[xx] = Find(fa[xx]), xx = belong[xx];
}
}
for (int i = 1; i < n; i++)
if (!Answer[i]) printf("-1\n"); else printf("%d\n", Answer[i]);
}