1. 程式人生 > 其它 >2022 ICPC 濟南站 L 題題解

2022 ICPC 濟南站 L 題題解

題意

給定一棵 \(n\) 個點有邊權的無根樹,有 \(q\) 次詢問,每次給定 \(l,r\),求

\[\min_{l\le u<v\le r} \{\operatorname{dist}(u,v)\}. \]

資料範圍:\(1\le n\le 2\times 10^5,1\le q\le 10^6\)

題解

仿照 CF765F 的做法,我們只保留若干對關鍵的點對。

考慮點分治,設當前分治中心是 \(u\)。考慮 \(u\) 的子樹中的一個點 \(x\),我們只需要找到那些 \(y\) 滿足 \(\operatorname{dist}(u,y)\le \operatorname{dist}(u,x)\)

\(y\)\(x\) 不在 \(u\) 的同一兒子的子樹內。

進一步考慮哪些 \(y\) 應當被保留。若存在兩個不同的點 \(y,z\) 使得它們都滿足上述條件,且 \(y<z<x\),那麼 \(\operatorname{dist}(y,z)=\operatorname{dist}(u,y)+\operatorname{dist}(u,z)\le \operatorname{dist}(u,y)+\operatorname{dist}(y,x)\)。此時 \((y,z)\) 這個點對一定比 \((y,x)\) 這個點對更優。

於是,對於每個 \(x\),設 \(S\)

是所有不與 \(x\)\(u\) 的同一兒子的子樹內,且 \(\operatorname{dist}(u,y)\le \operatorname{dist}(u,x)\)\(y\) 構成的集合,我們只需要保留 \(x\)\(S\) 中的前驅後繼即可。這樣,保留的點對數量是 \(O(n\log n)\) 的。

我們還可以發現,假如去掉“不能在同一兒子的子樹內”這個限制,若找到了在同一子樹內的點對 \((x,y)\),那麼說明其中一個點一定比另一個點更優。所以這個限制是可以去掉的。所以,對於每個分治中心 \(u\),我們直接統計出 \(u\) 子樹內的所有點的深度,排序後順著做一遍單調棧找到所有點的前驅,再倒著做一遍單調棧找到後繼,即可在 \(O(n\log n)\)

時間內找出所有關鍵點對。最後做一遍掃描線回答詢問即可。時間複雜度 \(O(n\log^2 n+q\log n)\)

程式碼

#include <bits/stdc++.h>
using namespace std;
#define For(Ti,Ta,Tb) for(int Ti=(Ta);Ti<=(Tb);++Ti)
#define Dec(Ti,Ta,Tb) for(int Ti=(Ta);Ti>=(Tb);--Ti)
#define Debug(...) fprintf(stderr,__VA_ARGS__)
using ll = long long;
const int N = 2e5 + 5, Q = 1e6 + 5;
int n, q, vis[N], siz[N], mxsiz[N];
vector<pair<int, int>> e[N];
void GetRoot(int u, int fa, int tot, int& rt) {
	siz[u] = 1, mxsiz[u] = 0;
	for (auto [v, w] : e[u]) if (v != fa && !vis[v]) {
		GetRoot(v, u, tot, rt);
		siz[u] += siz[v], mxsiz[u] = max(mxsiz[u], siz[v]);
	}
	mxsiz[u] = max(mxsiz[u], tot - siz[u]);
	if (!rt || mxsiz[u] < mxsiz[rt]) rt = u;
}
vector<pair<int, ll>> dis[N], dep;
void GetDep(int u, int fa, ll curd) {
	dep.emplace_back(u, curd);
	for (auto [v, w] : e[u]) if (v != fa && !vis[v]) {
		GetDep(v, u, curd + w);
	}
}
void Solve(int u) {
	int rt = 0; GetRoot(u, 0, siz[u], rt), u = rt, vis[u] = 1;
	dep.clear(), GetDep(u, 0, 0);
	sort(dep.begin(), dep.end());
	static int stk[N];
	for (int i = 0, tp = 0; i<int(dep.size()); ++i) {
		auto [v, w] = dep[i];
		while (tp && dep[stk[tp]].second > w) --tp;
		if (tp) dis[v].emplace_back(dep[stk[tp]].first, w + dep[stk[tp]].second);
		stk[++tp] = i;
	}
	for (int i = int(dep.size()) - 1, tp = 0; ~i; --i) {
		auto [v, w] = dep[i];
		while (tp && dep[stk[tp]].second > w) --tp;
		if (tp) dis[dep[stk[tp]].first].emplace_back(v, w + dep[stk[tp]].second);
		stk[++tp] = i;
	}
	for (auto [v, w] : e[u]) if (!vis[v]) Solve(v);
}
vector<pair<int, int>> qry[N];
struct BIT {
	ll t[N];
	void Upd(int p, ll x) { for (; p <= n; p += p & -p) t[p] = min(t[p], x); }
	ll Query(int p) { ll res = 1LL << 60; for (; p; p &= p - 1) res = min(res, t[p]); return res; }
}bit;
ll ans[Q];
int main() {
	ios::sync_with_stdio(false), cin.tie(nullptr);
	cin >> n;
	For(i, 1, n - 1) {
		int u, v, w; cin >> u >> v >> w;
		e[u].emplace_back(v, w), e[v].emplace_back(u, w);
	}
	siz[1] = n, Solve(1);
	cin >> q;
	For(i, 1, q) {
		int l, r; cin >> l >> r;
		qry[r].emplace_back(l, i);
	}
	memset(bit.t, 0x3f, sizeof(bit.t));
	For(r, 1, n) {
		for (auto [l, w] : dis[r]) bit.Upd(n - l + 1, w);
		for (auto [l, id] : qry[r]) {
			if (l == r) ans[id] = -1;
			else ans[id] = bit.Query(n - l + 1);
		}
	}
	For(i, 1, q) cout << ans[i] << '\n';
	return 0;
}