bzoj3924: [Zjoi2015]幻想乡战略游戏(动态点分治 + 欧拉序ST表求LCA)

题目:
在这里插入图片描述

题解:大坑啊,来自巨佬的题解 https://www.cnblogs.com/bztMinamoto/p/9489473.html
题意转化后就是求带权的重心,考虑暴力的做法:设原树的树根 r t rt rt s u m r t sum_{rt} sumrt为 rt的所有子树的点权的和, d i s r t dis_{rt} disrt为rt子树的点到 r t rt rt 的距离和(题目就是要求 d i s dis dis最小的那个点)。我们知道重心若不是根结点rt,必然在 rt的某一个子树中(废话),考虑移动到 rt 以 v 为根结点的子树,如果将重心移到 v,v 子树内的点对答案的贡献减少了 dis(rt,v),总共减少了 s u m r t ∗ d i s ( r t , v ) sum_rt * dis(rt,v) sumrtdis(rt,v),而其它的点的贡献都增加了 d i s ( r t , v ) dis(rt,v) dis(rt,v),总共增加了 ( t o t − s u m v ) ∗ d i s ( r t , v ) (tot - sum_v) * dis(rt,v) (totsumv)dis(rt,v),总的改变量就是 d i s ( r t , v ) ∗ ( t o t − s u m v − s u m v ) dis(rt,v) * (tot - sum_v - sum_v) dis(rt,v)(totsumvsumv)。观察到如果 v点更优,那么 s u m v > t o t 2 sum_v > \frac{tot}{2} sumv>2tot,显然 r t rt rt的众多儿子中最多只有一个优秀的儿子能满足这条件,对 v 也是一样,所以可以暴力的沿着树链往下跑最后找到最优的那个点。修改点权暴力维护需要 O(n)的时间,每一次查找最优答案也是O(n)的时间,T飞。

考虑如何优化:如果每次朝点分治的分治重心移动,维护的话也沿着重心向上跳更新,可以把复杂度将到 l o g n logn logn,引入动态点分治的思想,构建点分树使得树形平衡从而降低复杂度。

回忆点分治的过程,构建点分树就是在分治时在重心处连边,使得树形转化成一棵相较平衡的树。

void divide(int u) {								//构建点分树 
	done[u] = true;
	for(int i = head[u]; i + 1; i = nxt[i]) {
		if(done[to[i]]) continue;
		tot = son[to[i]];root = 0;
		getroot(to[i],-1);
		g[u].push_back(pii(root,to[i]));fa[root] = u;
		divide(root);
	}
}

分治函数中倒数第二行代码就是在重心处建边。
构建完点分树后复杂度已经得到了保证,就可以开始思考需要维护哪些信息,如何维护。
笨比不能一两句讲清楚,详细还得看大佬的博客(在上方)

(不理解为什么寻找带权重心时要在原树边判断是否要移动)

#include<bits/stdc++.h>
using namespace std;
const int maxn = 3e5 + 10;
const int maxm = 1e6 + 10;
#define pii pair<int,int>
#define fir first
#define sec second
typedef long long ll;
int n,q,rt;
int cnt,head[maxm],nxt[maxm],to[maxm],w[maxm];		//原图 
int son[maxn],f[maxn],root,tot;
bool done[maxn];
vector<pii> g[maxn]; 							//点分树
int fa[maxn];
int logn[maxn],bin[100],mx;
int dfn[maxn],num;
ll d[maxn],st[maxn][22];
ll v1[maxn],v2[maxn],v3[maxn];
void init() {
	fill(head,head + n + 1,-1);
	fill(fa,fa + n + 1,0);
	fill(v1,v1 + n + 1,0);
	fill(v2,v2 + n + 1,0);
	fill(v3,v3 + n + 1,0);
	fill(d,d + n + 1,0);
	num = mx = tot = cnt = root = 0;
} 
void add(int u,int v,int wi) {
	to[cnt] = v;
	w[cnt] = wi;
	nxt[cnt] = head[u];
	head[u] = cnt++;
}
void dfs(int u,int s) {
	dfn[u] = ++num;st[num][0] = d[u];
	for(int i = head[u]; i + 1; i = nxt[i]) {
		if(to[i] == s) continue;
		d[to[i]] = d[u] + w[i];
		dfs(to[i],u);
		st[++num][0] = d[u];
	}
}
ll lca(int a,int b) {
	if(dfn[a] > dfn[b]) swap(a,b);
	int k = logn[dfn[b] - dfn[a] + 1];
	return min(st[dfn[a]][k],st[dfn[b] - bin[k] + 1][k])<<1;
}
ll dis(int a,int b) {
	return d[a] + d[b] - lca(a,b);
}
void getroot(int u,int s) {
	son[u] = 1;f[u] = 0;
	for(int i = head[u]; i + 1; i = nxt[i]) {
		if(to[i] == s || done[to[i]]) continue;
		getroot(to[i],u);
		son[u] += son[to[i]];
		f[u] = max(f[u],son[to[i]]);
	}
	f[u] = max(f[u],tot - son[u]);
	if(!root || f[root] > f[u]) root = u;
}
void divide(int u) {								//构建点分树 
	done[u] = true;
	for(int i = head[u]; i + 1; i = nxt[i]) {
		if(done[to[i]]) continue;
		tot = son[to[i]];root = 0;
		getroot(to[i],-1);
		g[u].push_back(pii(root,to[i]));fa[root] = u;
		divide(root);
	}
}
void upd(int u,int val) {
	v3[u] += val;
	for(int p = u; fa[p]; p = fa[p]) {
		ll dist = dis(fa[p],u) * val;
		v1[fa[p]] += dist;
		v2[p] += dist;
		v3[fa[p]] += val;
	}
}
ll cal(int u) {
	ll ans = v1[u];
	for(int p = u; fa[p]; p = fa[p]) {
		ll dist = dis(fa[p],u);
		ans += v1[fa[p]] - v2[p];
		ans += (v3[fa[p]] - v3[p]) * dist;
	}
	return ans;
}
ll qry(int u) {
	ll ans = cal(u);
	for(auto it : g[u]) {
		ll tmp = cal(it.sec);
		if(tmp < ans) return qry(it.fir);
	}
	return ans;
}
int main() {
	scanf("%d%d",&n,&q);
	init();
	while((1 << (mx + 1)) <= (n << 1)) mx++;
	bin[0] = 1;logn[0] = -1;
	for(int i = 1; i <= mx; i++) bin[i] = bin[i - 1] << 1;
	for(int i = 1; i <= (n << 1); i++) logn[i] = logn[i >> 1] + 1;	
	for(int i = 1; i < n; i++) {
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w);add(v,u,w);
	}
	dfs(1,-1);
	for(int i = 1; i <= mx; i++)
		for(int j = 1;j + bin[i] - 1 <= num; j++)
			st[j][i] = min(st[j][i - 1],st[j + bin[i - 1]][i - 1]);
	root = 0;tot = n;getroot(1,-1);rt = root;
	divide(root);
	for(int i = 1; i <= q; i++) {
		int u,e;
		scanf("%d%d",&u,&e);
		upd(u,e);
		printf("%lld\n",qry(rt));
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值