标签：200001 le 进阶 剖分 int ybtoj father son now

三只企鹅

题目链接：ybtoj高效进阶 21270

题目大意

给你一棵树，然后要你支持一些操作。
给一个点的权值加一（一开始都是 0），计算所有点到一个点的距离乘各自点的权值。

思路

考虑把每个距离拆成 \(deg_x+deg_y-2deg_{lca}\)。

然后不难发现就第三项比较难搞。
考虑这么一种计算方法，在放点的时候，把它到根节点的路径上的边都加一，然后询问的时候它的根节点的路径的值的和就是第三项，感性理解即可看出你找到的就是 lca 到根的路径。

然后这个加的过程和查的过程可以用树链剖分和线段树实现。

代码

#include<cstdio>
#include<iostream>
#define ll long long

using namespace std;

struct node {
	int x, to, nxt;
}e[400001];
int n, m, le[200001], KK, tmpp;
int x, y, op, z, deg[200001], dy[200001];
int fa[200001], son[200001], top[200001];
int sz[200001], dfn[200001], cnt;
ll lsum, tmp[200001], degtmp[200001];

void add(int x, int y, int z) {
	e[++KK] = (node){z, y, le[x]}; le[x] = KK;
	e[++KK] = (node){z, x, le[y]}; le[y] = KK;
}

void dfs1(int now, int father) {//树链剖分预处理
	deg[now] = deg[father] + 1;
	fa[now] = father;
	sz[now] = 1;
	
	for (int i = le[now]; i; i = e[i].nxt)
		if (e[i].to != father) {
			tmp[e[i].to] = e[i].x;
			degtmp[e[i].to] = degtmp[now] + tmp[e[i].to];
			dfs1(e[i].to, now);
			sz[now] += sz[e[i].to];
			if (sz[e[i].to] > sz[son[now]]) son[now] = e[i].to;
		}
}

void dfs2(int now, int father) {
	dfn[now] = ++tmpp;
	dy[tmpp] = now;
	
	if (son[now]) {
		top[son[now]] = top[now];
		dfs2(son[now], now);
	}
	for (int i = le[now]; i; i = e[i].nxt)
		if (e[i].to != father && e[i].to != son[now]) {
			top[e[i].to] = e[i].to;
			dfs2(e[i].to, now);
		}
}

struct XDtree {//线段树
	ll a[800001], sum[800001];
	ll lzy[800001];
	
	void up(int now) {
		a[now] = a[now << 1] + a[now << 1 | 1];
		sum[now] = sum[now << 1] + sum[now << 1 | 1];
	}
	
	void down(int now) {
		if (!lzy[now]) return ;
		sum[now << 1] += a[now << 1] * lzy[now];
		sum[now << 1 | 1] += a[now << 1 | 1] * lzy[now];
		lzy[now << 1] += lzy[now];
		lzy[now << 1 | 1] += lzy[now];
		lzy[now] = 0;
	}
	
	void build(int now, int l, int r) {
		if (l == r) {
			a[now] = tmp[dy[l]];
			return ;
		}
		
		int mid = (l + r) >> 1;
		build(now << 1, l, mid);
		build(now << 1 | 1, mid + 1, r);
		up(now);
	}
	
	void insert(int now, int l, int r, int L, int R, ll t) {
		if (L <= l && r <= R) {
			sum[now] += a[now] * t;
			lzy[now] += t;
			return ;
		}
		
		down(now);
		int mid = (l + r) >> 1;
		if (L <= mid) insert(now << 1, l, mid, L, R, t);
		if (mid < R) insert(now << 1 | 1, mid + 1, r, L, R, t);
		up(now);
	}
	
	ll query(int now, int l, int r, int L, int R) {
		if (L <= l && r <= R) {
			return sum[now];
		}
		
		down(now);
		int mid = (l + r) >> 1;
		ll re = 0;
		if (L <= mid) re += query(now << 1, l, mid, L, R);
		if (mid < R) re += query(now << 1 | 1, mid + 1, r, L, R);
		return re;
	}
}T;

int main() {
//	freopen("express.in", "r", stdin);
//	freopen("express.out", "w", stdout);
	
	scanf("%d %d", &n, &m);
	for (int i = 1; i < n; i++) {
		scanf("%d %d %d", &x, &y, &z);
		add(x, y, z);
	}
	
	dfs1(1, 0);
	top[1] = 1;
	dfs2(1, 0);
	
	T.build(1, 1, n);
	while (m--) {
		scanf("%d %d", &op, &x);
		if (op == 1) {
			lsum += degtmp[x]; cnt++;
			while (x) {
				T.insert(1, 1, n, dfn[top[x]], dfn[x], 1);
				x = fa[top[x]];
			}
		}
		if (op == 2) {
			ll re = lsum + 1ll * cnt * degtmp[x];
			while (x) {
				re -= 2ll * T.query(1, 1, n, dfn[top[x]], dfn[x]);
				x = fa[top[x]];
			}
			printf("%lld\n", re);
		}
	}
	
	return 0;
}

标签：200001,le,进阶,剖分,int,ybtoj,father,son,now
来源： https://www.cnblogs.com/Sakura-TJH/p/YBTOJ_GXJJ_21270.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。