1. 程式人生 > >Jzoj P5914 盟主的憂慮___思維+並查集

Jzoj P5914 盟主的憂慮___思維+並查集

題目大意:

給出一棵有NN個節點的樹的N1N-1條邊,它們通過這些邊不需要消耗費用,然後給出MM條密道,需要花費wiw_i經過,將一開始的N1N-1條邊分別斷掉,然後每次求出兩個點斷掉以後通過剩下的邊和密道使這兩點連通花費的最少費用,無法連通就是-1。

N,M<=100,000N,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]);
}