算法笔记-线段树合并

最新推荐文章于 2024-04-28 00:34:31 发布
最新推荐文章于 2024-04-28 00:34:31 发布
阅读量931 收藏 1
点赞数 2
文章标签： 算法 笔记 图论 数据结构
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/m0_53899788/article/details/131036731
版权

线段树合并

前置知识：权值线段树、动态开点

将两棵线段树的信息合并成一棵线段树。
可以新建一颗线段树保存原来两颗线段树的信息，也可以将第二棵线段树维护的信息加到第一棵线段树上。

前者的空间复杂度较高，如果合并之前的线段树不会再用到的话，可以将第二颗线段树的信息加到第一棵线段树上。

P4556 [Vani有约会]雨天的尾巴 /【模板】线段树合并

题意：

一棵树有 n n n 个点。每次操作 ( x , y , z ) (x,y,z) (x,y,z) 在路径 ( x , y ) (x,y) (x,y) 上的每一个点放一个救济粮 z z z。询问每个点存放最多的是哪种救济粮

解析：

对于树上一条路径上的点进行相同的操作，可以想到树上差分。

然后统计每个点最多的东西，可以用权值线段树维护每种救济粮的数目。

因为将发放救济粮转化成树上差分，求答案的时候需要合并，所以从下向上合并线段树。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef double db;
#define fi first
#define se second
#define debug(x) cerr << #x << ": " << (x) << endl
#define rep(i, a, b) for(int i = (a); i <= (b); i++)
const int maxn = 1e5+10;
const int maxm = 1e5+10;
const int INF = 0x3f3f3f3f;
typedef pair<int, int> pii;


int head[maxn], tot;
struct edge{
	int to, nxt;
}e[maxn << 1];
struct node{
	int mcnt, id;
	int ls, rs;
}t[maxn * 60];
struct query{
	int x, y, z;
}q[maxn];

int cnt;
int n, m, MAX;
int rt[maxn], ans[maxn];
void add(int a, int b){
	e[++tot].nxt = head[a];
	e[tot].to = b;
	head[a] = tot;
}
int dep[maxn], siz[maxn], top[maxn], son[maxn], fa[maxn];
void dfs1(int u, int p){
	dep[u] = dep[p] + 1;
	siz[u] = 1;
	fa[u] = p;
	for(int i = head[u]; i; i = e[i].nxt){
		int v = e[i].to;
		if(v == p)
			continue;
		dfs1(v, u);
		siz[u] += siz[v];
		if(siz[v] > siz[son[u]])
			son[u] = v;
	}
}
void dfs2(int u, int tp){
	top[u] = tp;
	if(son[u])
		dfs2(son[u], tp);
	for(int i = head[u]; i; i = e[i].nxt){
		int v = e[i].to;
		if(v == fa[u] || v == son[u])
			continue;
		dfs2(v, v);
	}
}
int LCA(int u, int v){
	while(top[u] != top[v]){
		if(dep[top[u]] < dep[top[v]])
			swap(u, v);
		u = fa[top[u]];
	}
	return dep[u] < dep[v] ? u : v;
}
void pushup(int k){
	if(t[t[k].ls].mcnt >= t[t[k].rs].mcnt){
		t[k].mcnt = t[t[k].ls].mcnt;
		t[k].id = t[t[k].ls].id;
	}
	else{
		t[k].mcnt = t[t[k].rs].mcnt;
		t[k].id = t[t[k].rs].id;
	}
}
void update(int &k, int l, int r, int pos, int v){
	if(k == 0)
		k = ++cnt;
	if(l == r && l == pos){
		t[k].mcnt += v;
		t[k].id = l;
		return;
	}
	int mid = (l+r) >> 1;
	if(pos <= mid)
		update(t[k].ls, l, mid, pos, v);
	else
		update(t[k].rs, mid+1, r, pos, v);
	pushup(k);
}
void merge(int &a, int b, int l, int r){
	if(!a || !b){
		a = (!a ? b : a);
		return;
	}	
	if(l == r){
		t[a].mcnt += t[b].mcnt;
		t[a].id = l;
		return;
	}
	int mid = (l+r) >> 1;
	merge(t[a].ls, t[b].ls, l, mid);
	merge(t[a].rs, t[b].rs, mid+1, r);
	pushup(a);
}
void dfs(int u){
	for(int i = head[u]; i; i = e[i].nxt){
		int v = e[i].to;
		if(v == fa[u])
			continue;
		dfs(v);
		merge(rt[u], rt[v], 1, MAX);
	}
	if(t[rt[u]].mcnt != 0)
		ans[u] = t[rt[u]].id;
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0); cout.tie(0);
	
	cin >> n >> m;
	for(int i = 1, u, v; i < n; i++){
		cin >> u >> v;
		add(u, v);
		add(v, u);
	}
	dfs1(1, 0);
	dfs2(1, 1);
	
	for(int i = 1; i <= m; i++){
		cin >> q[i].x >> q[i].y >> q[i].z;
		MAX = max(MAX, q[i].z);
	}
	
	for(int i = 1; i <= m; i++){
		int u = q[i].x, v = q[i].y;
		int w = q[i].z;
		int lca = LCA(u, v);
		update(rt[u], 1, MAX, w, 1);
		update(rt[v], 1, MAX, w, 1);
		update(rt[lca], 1, MAX, w, -1);
		if(fa[lca])
			update(rt[fa[lca]], 1, MAX, w, -1);
	}
	dfs(1);
	for(int i = 1; i <= n; i++)
		cout << ans[i] << endl;
	
	return 0;
}


P1600 [NOIP2016 提高组] 天天爱跑步

题意：

一棵 n n n 节点的树，有 m m m 条路径，每个节点有参数 w i w_i wi​。询问有多少条路径的第 w i + 1 w_i+1 wi​+1 个点是节点 i i i。

解析：

对于每条路径 ( s , t ) (s, t) (s,t)，可以分成两条路径 ( s , l c a ) (s,lca) (s,lca) 和 ( l c a , t ) (lca,t) (lca,t)，如果模拟这个过程的话，时间复杂度为 O ( n m ) O(nm) O(nm) 不能接受。

换个角度考虑，对于每个点，有多少条路径会对该点产生贡献。

设节点 i i i 的深度为 d e p i dep_i depi​

设路径 ( s , t ) (s,t) (s,t) 对节点 u u u 产生贡献。

  • u u u 在 ( s , l c a ) (s,lca) (s,lca) 上。 d e p s − d e p u = w u dep_s-dep_u = w_u deps​−depu​=wu​
  • u u u 在 ( l c a , t ) (lca, t) (lca,t) 上。 d e p s + d e p u − 2 d e p l c a = w u dep_s+dep_u-2dep_{lca} = w_u deps​+depu​−2deplca​=wu​

即满足条件的路径会对节点 u u u 有贡献。

考虑树上差分，在 s s s 插入 d e p s dep_s deps​，在 t t t 处插入 2 d e p l c a − d e p s 2dep_{lca}-dep_s 2deplca​−deps​，在 l c a lca lca 处插入 − d e p s -dep_s −deps​，在 f a ( l c a ) fa(lca) fa(lca) 处插入 d e p s − 2 d e p l c a dep_s-2dep_{lca} deps​−2deplca​。后两者可以交换，然后线段树合并即可。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef double db;
#define fi first
#define se second
#define debug(x) cerr << #x << ": " << (x) << endl
#define rep(i, a, b) for(int i = (a); i <= (b); i++)
const int maxn = 3e5+10;
const int maxm = 1e5+10;
const int INF = 0x3f3f3f3f;
typedef pair<int, int> pii;

int head[maxn], cnt;
struct edge{
	int to, nxt;
}e[maxn << 1];
struct node{
	int v;
	int ls, rs;
}t[maxn * 60];
int n, m;
int rt[maxn], MAX, ans[maxn], w[maxn], tot;
void add(int a, int b){
	e[++cnt].nxt = head[a];
	e[cnt].to = b;
	head[a] = cnt;
} 
int siz[maxn], son[maxn], dep[maxn], fa[maxn], top[maxn];
void dfs1(int u, int p){
	dep[u] = dep[p]+1;
	siz[u] = 1;
	fa[u] = p;
	for(int i = head[u]; i; i = e[i].nxt){
		int v = e[i].to;
		if(v == p)
			continue;
		dfs1(v, u);
		siz[u] += siz[v]; 
		if(siz[v] > siz[son[u]])
			son[u] = v;
	}
}
void dfs2(int u, int tp){
	top[u] = tp;
	if(son[u])
		dfs2(son[u], tp);
	for(int i = head[u]; i; i = e[i].nxt){
		int v = e[i].to;
		if(v == son[u] || v == fa[u])
			continue;
		dfs2(v, v);
	}
}
int LCA(int u, int v){
	while(top[u] != top[v]){
		if(dep[top[u]] < dep[top[v]])
			swap(u, v);
		u = fa[top[u]];
	}
	return dep[u] < dep[v] ? u : v;
}
void update(int &k, int l, int r, int pos, int v){
	 if(k == 0)
	 	k = ++tot;
	if(l == r && l == pos){
		t[k].v += v;
		return;
	}
	int mid = (l+r) >> 1;
	if(pos <= mid)
		update(t[k].ls, l, mid, pos, v);
	else
		update(t[k].rs, mid+1, r, pos, v);
	return;
}
void merge(int &a, int b, int l, int r){
	if(!a || !b){
		a = (!a ? b : a);
		return;
	}
	if(l == r){
		t[a].v += t[b].v;
		return;
	}
	int mid = (l+r) >> 1;
	merge(t[a].ls, t[b].ls, l, mid);
	merge(t[a].rs, t[b].rs, mid+1, r);
}
int query(int k, int l, int r, int pos){
	if(!k)
		return 0;
	if(l == r)
		return t[k].v;
	int mid = (l+r) >> 1;
	if(pos <= mid)
		return query(t[k].ls, l, mid, pos);
	else
		return query(t[k].rs, mid+1, r, pos);
}
void dfs(int u){
	for(int i = head[u]; i; i = e[i].nxt){
		int v = e[i].to;
		if(v == fa[u])
			continue;
		dfs(v);
		merge(rt[u], rt[v], 1, MAX);
	}
	if(w[u] && n + dep[u] + w[u] <= MAX)
		ans[u] += query(rt[u], 1, MAX, n + dep[u] + w[u]);
	ans[u] += query(rt[u], 1, MAX, n + dep[u] - w[u]);
	
}

int main(){
	ios::sync_with_stdio(false);
	cin.tie(0); cout.tie(0);
	
	cin >> n >> m;
	MAX = n << 1;
	int u, v;
	for(int i = 1; i < n; i++){
		cin >> u >> v;
		add(u, v);
		add(v, u);
	}
	dfs1(1, 0);
	dfs2(1, 1);
	
	for(int i = 1; i <= n; i++)
		cin >> w[i];
	for(int i = 1; i <= m; i++){
		cin >> u >> v;
		int lca = LCA(u, v);
		update(rt[u], 1, MAX, n + dep[u], 1);
		update(rt[v], 1, MAX, n + dep[lca] * 2 - dep[u], 1);
		update(rt[lca], 1, MAX, n + dep[u], -1);
		update(rt[fa[lca]], 1, MAX, n + dep[lca] * 2 - dep[u], -1);
	}
	dfs(1);
	for(int i = 1; i <= n; i++)
		cout << ans[i] << " ";
	cout << endl;
	return 0;
}


P3224 [HNOI2012]永无乡

题意：

有 n n n 个节点，每个节点有互不相同的重要程度。两种操作：

  • 在 ( x , y ) (x,y) (x,y) 之间建桥
  • 询问与节点 x x x 所在连通块中重要程度排名第 k k k 小的节点编号

解析：

查询第 k k k 小，考虑权值线段树；维护联通性，考虑并查集。在合并两个连通块时，也合并权值线段树

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef double db;
#define fi first
#define se second
#define debug(x) cerr << #x << ": " << (x) << endl
#define rep(i, a, b) for(int i = (a); i <= (b); i++)
const int maxn = 3e5+10;
const int maxm = 1e5+10;
const int INF = 0x3f3f3f3f;
typedef pair<int, int> pii;

struct node{
	int sum, id;
	int ls, rs;
}t[maxn * 60];
int rt[maxn];
int tot, n, m, q;
int fa[maxn];
int find(int x){
	return fa[x] == x ? x : fa[x] = find(fa[x]);
}
int pushup(int k){
	t[k].sum = t[t[k].ls].sum + t[t[k].rs].sum;
}
void update(int &k, int l, int r, int pos, int idx){
	if(k == 0)
		k = ++tot;
	if(l == r){
		t[k].sum++;
		t[k].id = idx;
		return;
	}
	int mid = (l+r) >> 1;
	if(pos <= mid)
		update(t[k].ls, l, mid, pos, idx);
	else
		update(t[k].rs, mid+1, r, pos, idx);
	pushup(k);
}
void merge(int &a, int b, int l, int r){
	if(!a || !b){
		a = (a == 0 ? b : a);
		return;
	}
	if(l == r){
		t[a].sum += t[b].sum;
		return;
	}
	int mid = (l+r) >> 1;
	merge(t[a].ls, t[b].ls, l, mid);
	merge(t[a].rs, t[b].rs, mid+1, r);
	pushup(a);
}
int query(int a, int k, int l, int r){
	if(t[a].sum < k || !a)
		return 0;
	if(l == r)
		return t[a].id;
	int mid = (l+r) >> 1;
	int ans = 0;
	if(k <= t[t[a].ls].sum)
		ans = query(t[a].ls, k, l, mid);
	else
		ans = query(t[a].rs, k-t[t[a].ls].sum, mid+1, r);
	return ans;
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0); cout.tie(0);
	
	cin >> n >> m;
	for(int i = 1, p; i <= n; i++){
		fa[i] = i;
		cin >> p;
		update(rt[i], 1, n, p, i);
	}
	for(int i = 1, x, y; i <= m; i++){
		cin >> x >> y;
		int fx = find(x);
		int fy = find(y);
		fa[fy] = fx;
		merge(rt[fx], rt[fy], 1, n);
	}
	cin >> q;
	string op;
	for(int i = 1, x, y; i <= q; i++){
		cin >> op >> x >> y;
		if(op == "B"){
			int fx = find(x);
			int fy = find(y);
			if(fx == fy)
				continue;
			fa[fy] = fx;
			merge(rt[fx], rt[fy], 1, n); 
		}
		else if(op == "Q"){
			int fx = find(x);
			int res = query(rt[fx], y, 1, n);
			if(res == 0)
				cout << -1 << endl;
			else
				cout << res << endl;
		}
	}	
	return 0;
}
hydqiln
关注
  • 2
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
算法-数据结构- 线段树.rar
09-16
算法-数据结构- 线段树.rar
数据结构课程设计-线段树
06-08
数据结构课程的课程设计
C语言算法题-线段树
08-09
Alice和Bob又在挑战智力，这次的模式是这样的，Alice写下若干个数给Bob，之后会发出若干指令，让Bob给某数加上一个数（A），或者给某数减去一个数（S），或者查询某两个数之间所有数字之和（Q），有时他们也身份互换，反应速度最快的人将得到“速算小能手“奖章。你有好办法么？ 输入：第一行为数的个数N，第二行是这N个数，从第三行开始表示每个操作，第一个字母代表操作的种类，A和S后面的两个数i,j分别表示给第i个数加上或减去j，Q i,j表示查询i和j之间数字之和。 输出：若干行，每行为一次查询的结果。 样例输入： 10 1 2 3 4 5 6 7 8 9 10 Q 1 3 A 3 6 Q 2 7 S 10 2 A 6 3 Q 3 10 样例输出： 6 33 59
线段树的合并
Coldfresh的博客
07-15 340
题目描述： 现在有一棵二叉树，所有非叶子节点都有两个孩子。在每个叶子节点上有一个权值(有n个叶子节点，满足这些权值为1..n的一个排列)。可以任意交换每个非叶子节点的左右孩子。 要求进行一系列交换，使得最终所有叶子节点的权值按照遍历序写出来，逆序对个数最少。 input： 第一行n 下面每行，一个数x 如果x==0，表示这个节点非叶子节点，递归地向下读入其左孩子和右孩子的信息， 如果...
【学习】线段树和并
yhf_2015
03-23 149
（摘自 黄嘉泰 的课件） 合并： 由线段树的定义我们不难写出下面的过程，来合并两棵代表范围相同的线段树 merge(a,b): 如果a,b中有一个不含任何元素，就返回另一个 如果a,b都是叶子，返回merge_leaf(a,b) 返回merge(a->l,b->l)与merge(a->r,b->r)连接成的树 由于a,b两棵树结构相同，上面的过程的正确性是显然的
线段树合并
Cauchy的博客
05-21 2805
文章目录前置芝士线段树合并概念merge题目描述输入格式输出格式代码 前置芝士 权值线段树 动态开点线段树 线段树合并的全称为动态开点权值线段树合并 加个链接 线段树合并 概念 顾名思义，就是建立一棵新的线段树保存原有的两颗线段树的信息。 merge 线段树合并所完成的操作就是：在相同区间内，对应位置相加。 合并方式主要如下：两个相同区间的节点 p{p}p 和 q{q}q，合并区间 [l,r]{[l,r]}[l,r]。 若 p{p}p 和 q{q}q 有一者为 0{0}0，则另一者为根。 若 l=
算法艺术-线段树入门
08-01
本着大家都能学习的精神，上传一些线段树的学习资料，本资料还是属于入门的，但是说的比较详细，大家配合着习题应该能较快掌握线段树
统计的力量-线段树全接触-张昆玮
05-10
统计的力量——线段树全接触_张昆玮
线段树合并&&启发式合并笔记
weixin_30437337的博客
07-13 176
这俩东西听起来很高端，实际上很好写，应用也很多~ 线段树合并 线段树合并，顾名思义，就是建立一棵新的线段树保存原有的两颗线段树的信息。 考虑如何合并，对于一个结点，如果两颗线段树都有此位置的结点，则直接合并两结点的信息（如维护最大值则取max，维护和则相加），然后递归处理左右子树； 若只有一个有，直接返回即可。 这样子做时间复杂度取决于重合节点个数，一次最坏复杂度是$O(nlogn)$，...
线段树合并——小学六年级必读经典
qq_52383696的博客
07-23 409
线段树合并 是对于权值线段树来说的。 前置知识： 权值线段树，动态开点线段树； 如果以上两个知识点都还没点亮的话，建议出门右转，我帮你 return（ bushi 那么我们就发车啦! 线段树合并，顾名思义，就是建立一颗新的线段树保留原本两个线段树的信息。 考虑 我们合并到线段树a,线段树b的pos位置上: 如果a有pos位，b没有，直接return a； 如果b有pos位，a没有，直接return b; 如果已经合并到叶子节点了，就直接把b上pos位的值加到a的pos位上，return a; 递归处理左右子
关于线段树合并
m0_57344422的博客
08-14 347
线段树合并
线段树 & 线段树合并优化 DP
RabbieWjy的博客
05-19 485
线段树 & 线段树合并优化 DP 补充了之前动态规划优化课件中的线段树优化部分（？）
线段树应用——线段树合并
Lucas_FC_的博客
07-28 324
线段树合并
Codeforces Round 941 (Div. 2) （A~D）
最新发布
weixin_61825750的博客
04-28 705
然后考虑若此时存在两个堆，那么对于Alice而言，其操作可以使得小的那个堆只剩下1，下一步Bob必须选1，然后Alice就胜利了。相反，若小的那个堆一开始就是1，那么Alice就输了。接下来推广到若干堆的情况：对于任意一方而言，若最小堆不是1，那么便可以控制胜利，于是我们只需要枚举到最小堆不是1的情况即可，然后看是谁在操作就是谁赢。思路：若要使得所有方格同色，关键在于四个角上的颜色，若对角上的颜色一样就能直接将所有格子变为同色。思路：首先处理 < k 的部分，我们可以通过二进制的思想来构造整数，例如。
廊桥分配csp-s线段树解法
11-17
廊桥分配问题是指给定一座长度为...使用线段树解决廊桥分配问题的时间复杂度为O(nlogn)，因为排序的时间复杂度为O(nlogn)，遍历人员列表的时间复杂度为O(n)，遍历线段树的时间复杂度为O(nlogn)。总的空间复杂度为O(n)。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值