1. 程式人生 > 實用技巧 >[BJWC2010]嚴格次小生成樹

[BJWC2010]嚴格次小生成樹

傳送門


這道題以前我沒做明白,今天再看看發現竟然挺好理解的。


首先我們求出最小生成樹,這樣我們只需要在原樹上改動一條邊就可以得到次小生成樹了。
具體怎麼改呢?已經有\(n - 1\)條邊參與構成最小生成樹了,我們列舉剩下的\(m - n + 1\)條邊。記這條邊的兩個端點為\(x\)\(y\),權值為\(w\),那麼我們只要嘗試用這條邊替換\(x\)\(y\)的樹上路徑的其中一條邊就行了。值得注意的是,\(w\)一定大於等於路徑上的任意一條邊,因為如果小於的話,就可以把那條邊刪了,把\(w\)加進來構成最小生成樹。


接下來只要做兩件事:記路徑上最大的邊權為\(w_{max}\),如果\(w > w_{max}\)

,那麼直接替換並嘗試更新答案;否則只可能\(w = w_{max}\),那麼就要用\(w\)替換路徑上的次大值。
這樣列舉\(m - n + 1\)條邊後,選擇替換後的最小值就是答案。


所以我們要做的就是怎麼求出路徑上的最小值和次小值。
樹上的路徑,自然會想到拆成\(x\)到lca,和lca到\(y\)的兩條路徑。那麼可以在求lca的倍增陣列的時候順便把最大值和次大值維護出來。
\(Max[0][i][x]\)表示\(x\)往上\(2 ^ i\)長度的路徑中的最大值,\(Max[1][i][x]\)為對應的次大值,現在考慮用\(i - 1\)來更新\(i\)
對於最大值來說,自然是

\[Max[0][i][x] = max \{Max[0][i - 1][x], Max[0][i - 1][fa[i - 1][x]] \} \]

對於次大值的維護,要比較兩個\(i-1\)段的最大值哪一個大:

\[\begin{align*} Max[1][i][x] = max \{Max[1][i - 1][x], Max[0][i - 1][fa[i - 1][x]] \}(Max[0][i-1][x] > Max[0][i - 1][fa[i - 1][x]]) \\ = max \{Max[0][i - 1][x], Max[1][i - 1][fa[i - 1][x]] \}(Max[0][i-1][x] > Max[0][i - 1][fa[i - 1][x]]) \\ = max \{Max[1][i - 1][x], Max[1][i - 1][fa[i - 1][x]] \}(Max[0][i-1][x] = Max[0][i - 1][fa[i - 1][x]]) \end{align*} \]

處理完這些後,我們在求lca的時候就可以仿照倍增的思路求出路徑上的最大值和次大值了。


程式碼實現推薦用一個pair存最大值和次大值。因為無論是預處理還是倍增的時候,求最大值和次大值的邏輯是一樣的,所以可以用一個函式封裝起來,返回pair型別。

#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cctype>
#include<vector>
#include<queue>
#include<assert.h>
#include<ctime>
using namespace std;
#define enter puts("") 
#define space putchar(' ')
#define Mem(a, x) memset(a, x, sizeof(a))
#define In inline
#define forE(i, x, y) for(int i = head[x], y; ~i && (y = e[i].to); i = e[i].nxt)
typedef long long ll;
typedef double db;
const ll INF = 1e18;
const db eps = 1e-8;
const int maxn = 1e5 + 5;
const int maxm = 3e5 + 5;
const int N = 18;
In ll read()
{
	ll ans = 0;
	char ch = getchar(), las = ' ';
	while(!isdigit(ch)) las = ch, ch = getchar();
	while(isdigit(ch)) ans = (ans << 1) + (ans << 3) + ch - '0', ch = getchar();
	if(las == '-') ans = -ans;
	return ans;
}
In void write(ll x)
{
	if(x < 0) x = -x, putchar('-');
	if(x >= 10) write(x / 10);
	putchar(x % 10 + '0');
}
In void MYFILE()
{
#ifndef mrclr
	freopen(".in", "r", stdin);
	freopen(".out", "w", stdout);
#endif
}

int n, m;
struct Edge
{
	int nxt, to, w;
}e[maxn << 1];
int head[maxn], ecnt = -1;
In void addEdge(int x, int y, int w)
{
	e[++ecnt] = (Edge){head[x], y, w};
	head[x] = ecnt;
}

int p[maxn];
In int Find(int x) {return x == p[x] ? x : p[x] = Find(p[x]);}
In void merge(int x, int y)
{
	int px = Find(x), py = Find(y);
	if(px == py) return;
	p[px] = py;
}

bool vis[maxm];
struct edges
{
	int x, y, w;
	In bool operator < (const edges& oth)const
	{
		return w < oth.w;
	}
}ed[maxm];

#define pr pair<int, int>
#define mp make_pair
#define F first
#define S second
In pr calc(pr a, pr b)
{
	if(a.F == b.F) return mp(a.F, max(a.S, b.S));
	else if(a.F > b.F) return mp(a.F, max(a.S, b.F));
	else return mp(b.F, max(a.F, b.S));
}

int dep[maxn], fa[N + 2][maxn];
pr Max[N + 2][maxn];
In void dfs(int now, int _f)
{
	for(int i = 1; (1 << i) <= dep[now]; ++i)
	{
		fa[i][now] = fa[i - 1][fa[i - 1][now]];
		Max[i][now] = calc(Max[i - 1][now], Max[i - 1][fa[i - 1][now]]);
	}
	forE(i, now, v)
	{ 
		if(v == _f) continue;
		dep[v] = dep[now] + 1, fa[0][v] = now;
		Max[0][v].F = e[i].w;
		dfs(v, now);
	}
}

In pr solve(int x, int y)
{
	pr ret = mp(0, 0);
	if(dep[x] < dep[y]) swap(x, y);
	for(int i = N; i >= 0; --i)
		if(dep[fa[i][x]] > dep[y])
		{
			ret = calc(ret, Max[i][x]);
			x = fa[i][x];
		}
	if(fa[0][x] == y) return ret;
	ret = calc(ret, Max[0][x]);
	x = fa[0][x];
	for(int i = N; i >= 0; --i)
		if((fa[i][x] ^ fa[i][y]) || !i)
		{
			ret = calc(ret, calc(Max[i][x], Max[i][y]));
			x = fa[i][x], y = fa[i][y];
		}
	return ret;
}

int main()
{
//	MYFILE();
	n = read(), m = read();
	Mem(head, -1), ecnt = -1;
	for(int i = 1; i <= n; ++i) p[i] = i;
	for(int i = 1; i <= m; ++i)
		ed[i].x = read(), ed[i].y = read(), ed[i].w = read();
	sort(ed + 1, ed + m + 1);
	ll sum = 0, ans = INF;
	for(int i = 1; i <= m; ++i)
	{
		int x = ed[i].x, y = ed[i].y, w = ed[i].w;
		if(Find(x) ^ Find(y))
		{
			merge(x, y), sum += w;
			addEdge(x, y, w), addEdge(y, x, w); 
			vis[i] = 1;
		}
	}
	dep[1] = 1, dfs(1, 0);
	for(int i = 1; i <= m; ++i)
		if(!vis[i])
		{
			pr tp = solve(ed[i].x, ed[i].y);
			if(ed[i].w > tp.F) ans = min(ans, sum - tp.F + ed[i].w);
			else if(ed[i].w == tp.F) ans = min(ans, sum - tp.S + ed[i].w); 
		}
	write(ans), enter; 
	return 0;	
}