可持久化数组(主席树)

可持久化数组引入:P3919 【模板】可持久化线段树 1(可持久化数组) - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

参考:P3919 【模板】可持久化线段树 1(可持久化数组) - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

视频讲解:【AgOHの数据结构】主席树_哔哩哔哩_bilibili 


 AC代码

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<stdlib.h>
#define ll long long

using namespace std;

const int maxn = 5e6 + 10;

struct zxs
{
	int l,r,v;	
}t[maxn * 20];

int a[maxn],node[maxn];//node[i]为i版本的根编号,刚开始编号为0 
int top;

int copy(int node)	//新建节点
{
	top++;
	t[top] = t[node];	//全部信息都传到新节点
	return top;
}

int build(int root,int l,int r)	//建树其实就是新建节点的过程
{
	root = ++top;
	if(l == r)
	{
		t[root].v = a[l];
		return root;
	}
	int m = (l+r)>>1;
	t[root].l = build(t[root].l,l,m);
	t[root].r = build(t[root].r,m+1,r);
	return root;
}

int update(int root,int l,int r,int loc,int val)
{
	root = copy(root);	//更新就要新建节点 
	if(l == r)
		t[root].v = val;
	else
	{
		int m = (l+r)>>1;
//如果要更新的树不是单独节点的树,逐层递归,寻找需要修改的节点以减少空间 
		if(loc <= m)
			t[root].l = update(t[root].l,l,m,loc,val); 
		else
			t[root].r = update(t[root].r,m+1,r,loc,val);
	}
	return root;
}

int query(int root,int l,int r,int loc)
{
	if(l == r)
		return t[root].v;
	int m = (l+r)>>1;
	if(loc <= m)
		return query(t[root].l,l,m,loc); 
	else
		return query(t[root].r,m+1,r,loc);
}

int main()
{
	int n,k,ver,op,loc,val,res;
	scanf("%d%d", &n, &k);
	for(int i=1;i<=n;i++)
		scanf("%d", &a[i]);
		
	node[0] = build(0,1,n);
	for(int i=1;i<=k;i++)
	{
		scanf("%d%d%d", &ver, &op, &loc);
		if(op == 1)	//修改 
		{
			scanf("%d", &val);
			node[i] = update(node[ver],1,n,loc,val);	//修改版本 
		}
		else	//查询 
		{
			res = query(node[ver],1,n,loc);
			printf("%d\n", res);
			node[i] = node[ver];	//保存版本 
		}
	}

	return 0;
	
} 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
持久化左偏树是一种支持历史版本查询的数据结构,它可以在不破坏原有数据结构的基础上,对其进行修改和查询。下面是一个简单的C++可持久化左偏树的实现: ```c++ #include <bits/stdc++.h> using namespace std; const int MAXN = 1e5 + 5; struct Node { int val, ls, rs; } t[MAXN * 40]; int n, m, cnt, root[MAXN]; int build(int l, int r) { int p = ++cnt; if (l == r) { t[p].val = 0; return p; } int mid = (l + r) >> 1; t[p].ls = build(l, mid); t[p].rs = build(mid + 1, r); return p; } int update(int pre, int l, int r, int x, int v) { int p = ++cnt; t[p] = t[pre]; if (l == r) { t[p].val += v; return p; } int mid = (l + r) >> 1; if (x <= mid) t[p].ls = update(t[pre].ls, l, mid, x, v); else t[p].rs = update(t[pre].rs, mid + 1, r, x, v); return p; } int query(int u, int v, int l, int r, int k) { if (l == r) return l; int mid = (l + r) >> 1; int cnt = t[t[v].ls].val - t[t[u].ls].val; if (cnt >= k) return query(t[u].ls, t[v].ls, l, mid, k); else return query(t[u].rs, t[v].rs, mid + 1, r, k - cnt); } int main() { scanf("%d%d", &n, &m); root[0] = build(1, n); for (int i = 1; i <= n; i++) { int x; scanf("%d", &x); root[i] = update(root[i - 1], 1, n, x, 1); } while (m--) { int l, r, k; scanf("%d%d%d", &l, &r, &k); printf("%d\n", query(root[l - 1], root[r], 1, n, k)); } return 0; } ``` 上述代码实现了一个可持久化左偏树,支持区间第k小查询。其中,build函数用于建立一棵空树,update函数用于在原有版本的基础上插入一个新节点,query函数用于查询区间第k小。在主函数中,我们首先建立一棵空树,然后依次插入每个节点,最后进行m次区间查询。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值