洛谷 P3919 【模板】可持久化数组(可持久化线段树/平衡树)(主席树)

这个题作为主席树的一道模板题,格式有问题,题目就不复制过来了
原题连接
solution:主席树即持久化线段树,如果不懂线段树建议先去看看我的线段树博客

#include <bits/stdc++.h>
using namespace std;

const int maxn = 1000001;
int top = 0, a[maxn], root[maxn];

struct xjp{
	int l, r, val;
}tree[maxn * 20];

//建立新节点
int colne(int node){
	++top;
	tree[top] = tree[node];
	return top;
}

//建树
int maketree(int node, int begin, int end)
{
	node = ++top;
	if (begin == end){
		tree[node].val = a[begin];
		return top;
	}
	int mid = (begin + end) >> 1;
	tree[node].l = maketree(tree[node].l, begin, mid);
	tree[node].r = maketree(tree[node].r, mid + 1, end);
	return node;
}

//更新
int update(int node, int begin, int end, int x, int val)
{
	node = colne(node);
	if (begin == end)tree[node].val = val;
	else{
		int mid = (begin + end) >> 1;
		if (x <= mid)tree[node].l = update(tree[node].l, begin, mid, x, val);
		else tree[node].r = update(tree[node].r, mid + 1, end, x, val);
	}
	return node;
}

//询问
int query(int node, int begin, int end, int x)
{
	if (begin == end)return tree[node].val;
	else {
		int mid = (begin + end) >> 1;
		if (x <= mid)return query(tree[node].l, begin, mid, x);
		else return query(tree[node].r, mid + 1, end, x);
	}
}

int main()
{
	int n, m;
	scanf("%d%d", &n, &m);
	for (int i = 1; i <= n; ++i)scanf("%d", &a[i]);
	root[0] = maketree(0, 1, n);
	for  (int i = 1; i <= m; ++i){
		int rt, mode, x, y;
		scanf("%d%d%d", &rt, &mode, &x);
		if (mode == 1){
			scanf("%d", &y);
			root[i] = update(root[rt], 1, n, x, y);
		}else{
			printf("%d\n", query(root[rt], 1, n, x));
			root[i] = root[rt];
		}
	}
	return 0;
}
全部评论

相关推荐

野猪不是猪🐗:这种直接口头上答应,骗面试,面完了直接拉黑,相当于给自己攒面经了(
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务