【BZOJ3991】【SDOI2015】寻宝游戏

最新推荐文章于 2020-04-13 11:22:55 发布

cz_xuyixuan

最新推荐文章于 2020-04-13 11:22:55 发布

阅读量285

点赞数

CC 4.0 BY-SA版权

分类专栏：【OJ】BZOJ 【类型】做题记录【数据结构】虚树

本文链接：https://blog.youkuaiyun.com/qq_39972971/article/details/79555905

【类型】做题记录同时被 3 个专栏收录

767 篇文章

订阅专栏

【OJ】BZOJ

397 篇文章

订阅专栏

【数据结构】虚树

14 篇文章

订阅专栏

本文介绍了一种使用LCA算法求解树上两点间最短路径的方法，并结合动态维护策略解决了一系列涉及节点插入删除的问题。通过实例演示了如何在不断变化的树结构中高效更新节点间的距离。

【题目链接】

点击打开链接

【思路要点】

补档博客，无题解。

【代码】

#include<bits/stdc++.h>
using namespace std;
#define MAXN	100005
#define MAXLOG	20
struct edge {int dest; long long len; };
struct info {int value; };
vector <edge> a[MAXN];
set <info> BBT;
long long dist[MAXN];
int timer, dfn[MAXN], depth[MAXN], father[MAXN][MAXLOG];
void init(int pos, int fa, int dep, long long length) {
	depth[pos] = dep;
	father[pos][0] = fa;
	dist[pos] = length;
	dfn[pos] = ++timer;
	for (int i = 1; i < MAXLOG; i++)
		father[pos][i] = father[father[pos][i - 1]][i - 1];
	for (unsigned i = 0; i < a[pos].size(); i++)
		if (a[pos][i].dest != fa) init(a[pos][i].dest, pos, dep + 1, length + a[pos][i].len);
} 
int lca(int x, int y) {
	if (depth[x] < depth[y]) swap(x, y);
	for (int i = MAXLOG - 1; i >= 0; i--)
		if (depth[father[x][i]] >= depth[y]) x = father[x][i];
	if (x == y) return x;
	for (int i = MAXLOG - 1; i >= 0; i--)
		if (father[x][i] != father[y][i]) {
			x = father[x][i];
			y = father[y][i];
		}
	return father[x][0];
}
long long length(int x, int y) {
	return dist[x] + dist[y] - dist[lca(x, y)] * 2;
}
bool operator < (info x, info y) {
	return dfn[x.value] < dfn[y.value];
}
bool operator > (info x, info y) {
	return dfn[x.value] > dfn[y.value];
}
int func(info tmp) {
	set <info> :: iterator tnp;
	tnp = BBT.end(); tnp--;
	if (*tnp < tmp) return (*tnp).value;
	tnp = BBT.lower_bound(tmp); tnp--;
	return (*tnp).value;
}
int main() {
	int n, m;
	scanf("%d%d", &n, &m);
	for (int i = 1; i < n; i++) {
		int x, y, z;
		scanf("%d%d%d", &x, &y, &z);
		a[x].push_back((edge){y, z});
		a[y].push_back((edge){x, z});
	}
	init(1, 0, 1, 0);
	long long ans = 0;
	for (int i = 1; i <= m; i++) {
		info tmp;
		scanf("%d", &tmp.value);
		if (BBT.empty()) {
			BBT.insert(tmp);
			printf("0\n");
			continue;
		}
		if (BBT.count(tmp)) {
			BBT.erase(tmp);
			if (BBT.empty()) {
				printf("0\n");
				continue;
			}
			set <info> :: iterator tnp;
			tnp = BBT.end(); tnp--;
			if (*BBT.begin() < tmp) ans -= length(func(tmp), tmp.value);
			else ans -= length((*tnp).value, tmp.value);
			if (*tnp > tmp) ans -= length((*BBT.lower_bound(tmp)).value, tmp.value);
			else ans -= length((*BBT.begin()).value, tmp.value);
			if (*BBT.begin() < tmp && *tnp > tmp) ans += length(func(tmp), (*BBT.lower_bound(tmp)).value);
			else if (*BBT.begin() < tmp) ans += length(func(tmp), (*BBT.begin()).value);
			     else ans += length((*tnp).value, (*BBT.lower_bound(tmp)).value);
		} else {
			if (BBT.empty()) {
				BBT.insert(tmp);
				printf("0\n");
				continue;
			}
			set <info> :: iterator tnp;
			tnp = BBT.end(); tnp--;
			if (*BBT.begin() < tmp) ans += length(func(tmp), tmp.value);
			else ans += length((*tnp).value, tmp.value);
			if (*tnp > tmp) ans += length((*BBT.lower_bound(tmp)).value, tmp.value);
			else ans += length((*BBT.begin()).value, tmp.value);
			if (*BBT.begin() < tmp && *tnp > tmp) ans -= length(func(tmp), (*BBT.lower_bound(tmp)).value);
			else if (*BBT.begin() < tmp) ans -= length(func(tmp), (*BBT.begin()).value);
			     else ans -= length((*tnp).value, (*BBT.lower_bound(tmp)).value);
			BBT.insert(tmp);
		}
		printf("%lld\n", ans);
	}
	return 0;
}