首页 > 其他分享 >题解 ABC294G【Distance Queries on a Tree】

题解 ABC294G【Distance Queries on a Tree】

时间:2023-03-21 20:55:06浏览次数:40  
标签:dep 题解 ll ABC294G fa dfn Queries operatorname lld

DFS 序树状数组。

不妨以 \(1\) 为根,设 \(\operatorname{dep}(u)\) 表示 \(u\) 到根路径的边权和,\(\operatorname{dis}(u,v)\) 表示 \(u,v\) 间路径的边权和,\(\operatorname{LCA}(u,v)\) 表示 \(u,v\) 的最近公共祖先。则显然有:\(\operatorname{dis}(u,v)=\operatorname{dep}(u)+\operatorname{dep}(v)-2\times\operatorname{dep}(\operatorname{LCA}(u,v))\)。只需要维护每个点的 \(\operatorname{dep}\)。

给一条边重新赋权值等价于给它的边权加上 \(\Delta w\),这一操作会导致这条边的子节点所在子树内每个点的 \(\operatorname{dep}\) 增加 \(\Delta w\)。使用 DFS 序即可将子树转化为区间,区间加、单点查只需要差分后树状数组维护即可。

时间复杂度 \(\mathcal O((n+q)\log n)\)。

// Problem: G - Distance Queries on a Tree
// Contest: AtCoder - AtCoder Beginner Contest 294
// URL: https://atcoder.jp/contests/abc294/tasks/abc294_g
// Memory Limit: 1024 MB
// Time Limit: 4000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

//By: OIer rui_er
#include <bits/stdc++.h>
#define rep(x,y,z) for(ll x=(y);x<=(z);x++)
#define per(x,y,z) for(ll x=(y);x>=(z);x--)
#define debug(format...) fprintf(stderr, format)
#define fileIO(s) do{freopen(s".in","r",stdin);freopen(s".out","w",stdout);}while(false)
#define likely(exp) __builtin_expect(!!(exp), 1)
#define unlikely(exp) __builtin_expect(!!(exp), 0)
using namespace std;
typedef long long ll;

mt19937 rnd(std::chrono::duration_cast<std::chrono::nanoseconds>(std::chrono::system_clock::now().time_since_epoch()).count());
ll randint(ll L, ll R) {
	uniform_int_distribution<ll> dist(L, R);
	return dist(rnd);
}

template<typename T> void chkmin(T& x, T y) {if(x > y) x = y;}
template<typename T> void chkmax(T& x, T y) {if(x < y) x = y;}

const ll N = 2e5+5;

ll n, m, fa[N][19], dis[N], sz[N], bottom[N], dfn[N], tms, val[N];
vector<tuple<ll, ll, ll>> e[N];

struct BIT {
	ll c[N];
	ll lowbit(ll x) {return x & (-x);}
	void add(ll x, ll k) {for(; x <= n; x += lowbit(x)) c[x] += k;}
	ll ask(ll x) {ll k = 0; for(; x; x -= lowbit(x)) k += c[x]; return k;}
}bit;

void dfs(ll u, ll f) {
	fa[u][0] = f;
	rep(i, 1, 18) fa[u][i] = fa[fa[u][i-1]][i-1];
	dis[u] = dis[f] + 1;
	dfn[u] = ++tms;
	sz[u] = 1;
	for(auto i : e[u]) {
		ll v, w, id;
		tie(v, w, id) = i;
		if(v != f) {
			dfs(v, u);
			sz[u] += sz[v];
			bottom[id] = v;
			bit.add(dfn[v], w);
			bit.add(dfn[v]+sz[v], -w);
		}
	}
}

ll LCA(ll u, ll v) {
	if(dis[u] < dis[v]) swap(u, v);
	per(i, 18, 0) if(dis[fa[u][i]] >= dis[v]) u = fa[u][i];
	if(u == v) return u;
	per(i, 18, 0) if(fa[u][i] != fa[v][i]) u = fa[u][i], v = fa[v][i];
	return fa[u][0];
}

int main() {
	scanf("%lld", &n);
	rep(i, 1, n-1) {
		ll u, v, w;
		scanf("%lld%lld%lld", &u, &v, &w);
		e[u].emplace_back(v, w, i);
		e[v].emplace_back(u, w, i);
		val[i] = w;
	}
	dfs(1, 0);
	for(scanf("%lld", &m); m; m--) {
		ll op, u, v;
		scanf("%lld%lld%lld", &op, &u, &v);
		if(op == 1) {
			ll bot = bottom[u];
			bit.add(dfn[bot], v-val[u]);
			bit.add(dfn[bot]+sz[bot], val[u]-v);
			val[u] = v;
		}
		else {
			ll lca = LCA(u, v);
			printf("%lld\n", bit.ask(dfn[u])+bit.ask(dfn[v])-2*bit.ask(dfn[lca]));
		}
	}
	return 0;
}

标签:dep,题解,ll,ABC294G,fa,dfn,Queries,operatorname,lld
From: https://www.cnblogs.com/ruierqwq/p/abc294g.html

相关文章

  • 【题解】CF487E Tourists / 圆方树
    概念圆方树是一种基于无向图构造的树。我们知道,圆方树最早是WC上提出的处理仙人掌的东西,用于将树上做法拓展到复杂度正确的仙人掌做法。但是一些关于点双有性质的题也......
  • CF123E maze 题解
    思考暴力:枚举起点和终点,再枚举每一种遍历序列得到答案。复杂度起飞。根据期望的可加性,我们无需硬着头皮统计每一条序列的贡献,而是把序列的贡献拆成遍历序列包含的边的贡献......
  • 蓝桥楼赛第9期-修复未正确实现的实验类 题解
    题目描述程序存放的位置/home/shiyanlou/lab.py;实验类名应该为Lab;实验对象中不能插入重复标签;Python中对象引用问题,尤其如复合对象list,dict,tuple的......
  • 【AT_abc294_g 题解】
    题意给定一颗\(n\)个节点的带权无向树。给出\(q\)个操作:1iw:把第\(i\)条边的边权变成\(w\)。2uv:求\(u\tov\)简单路径的边权和。解法根据树上差分。......
  • ABC288Ex 题解
    题意传送门给定\(n,m,x\),询问有多少个长度为\(n\)的非负整数序列满足以下条件:\(0\lea_1\lea_2\le\dots\lea_n\lem\)\(a_1\oplusa_2\oplus\dots\oplusa_n=x\)......
  • 问题解决01:默认不执行.ps1文件 - 无法双击.ps1文件
    默认不允许执行.ps1文件扩展名为.ps1的文件是用PowerShell写好的脚本文件。在Windows系统中,默认情况下是不允许执行.ps1文件。想双击一下执行.ps1文件?双击ps1文件很有......
  • abc294G
    UpdG看上好模板的样子,果然是个模板题好题,首先考虑这张图的\(Euler\Tour\),简单点说,就是dfs一遍,把每个点入栈出栈顺序存起来,举个例子·21223这棵树的......
  • 【题解】CF1034E
    题目描述给定\(n\)和长度为\(2^n\)的数列\(a_{0},a_{1}...a_{2^n-1}\)和\(b_{0},b_1...b_{2^n-1}\),保证每个元素的值属于\([0,3]\)生成序列\(c\),对于\(......
  • 【题解】CF889E
    题目描述\[f(x,n)=x\moda_n\]\[f(x,i)=(x\moda_i)+f(x\moda_i,i+1)\]给出a序列,当x取遍所有非负整数时\(f(x,1)\)的最大值。题解首先注意到\(a_i\)只......
  • 【题解】CF1368E
    题目描述有一个由\(n\)个点\(m\)条边组成的有向无环图,每个点出度至多为2。您需要标记一些点(不超过\(\frac{4}{7}n\)个)。标记一个点\(u\)将会删除所有与\(u\)连......