luogu P3919 [模板]可持久化数组(可持久化线段树/平衡树)(主席树)

luogu P3919 [模板]可持久化数组(可持久化线段树/平衡树) 题目

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<iomanip>
#include<algorithm>
#include<ctime>
#include<queue>
#include<stack>
#define rg register
#define il inline
#define lst long long
#define N 1000050
using namespace std; int n,Q,cnt;
int val[N];
int root[N];
struct TREE{
int ls,rs,v;
}ljl[N*]; il int read()
{
rg int s=,m=;rg char ch=getchar();
while(ch!='-'&&(ch<''||ch>''))ch=getchar();
if(ch=='-')m=-,ch=getchar();
while(ch>=''&&ch<='')s=(s<<)+(s<<)+ch-'',ch=getchar();
return s*m;
} il void build(rg int &now,rg int le,rg int ri)
//现在所在的点(可往回传参), 左端点, 右端点
{
ljl[++cnt]=ljl[now];//新开一个点……
now=cnt;//保证now指的是当点这个点,因为还要传回去赋值给“爸爸的左/右儿子”……具体看后面的递归build
if(le==ri){ljl[now].v=val[le];return;}//如果到单点赋值就ojbk了(同线段树)
rg int mid=(le+ri)>>;
build(ljl[now].ls,le,mid),build(ljl[now].rs,mid+,ri);
//建造左儿子和右儿子,本节点向他们的指针在递归函数往回传参时会赋值(一切都源于一个美丽的“ & ”)
} void Modify(rg int &now,rg int le,rg int ri,rg int kk,rg int x)//很像build
//现在所在的点(可往回传参),左端点,右端点,要修改的点编号,修改后的值
{
ljl[++cnt]=ljl[now];//又要开点了……
now=cnt;//保证now指的是当点这个点,因为还要传回去赋值给“爸爸的左/右儿子”……具体看后面的递归Modify
if(le==ri){ljl[now].v=x;return;}//照样赋值
rg int mid=(le+ri)>>;
if(kk<=mid)Modify(ljl[now].ls,le,mid,kk,x);
else Modify(ljl[now].rs,mid+,ri,kk,x);
//kk在mid左边,就建左孩子,否则建右孩子……需要模拟哦……
} int Query(rg int now,rg int le,rg int ri,rg int kk)
{
if(le==ri)return ljl[now].v;
rg int mid=(le+ri)>>;
if(kk<=mid)return Query(ljl[now].ls,le,mid,kk);
else return Query(ljl[now].rs,mid+,ri,kk);
} int main()
{
n=read(),Q=read();
for(rg int i=;i<=n;++i)val[i]=read();
build(root[],,n);//先按原来的值建一棵线段树在0号根上
for(rg int i=;i<=Q;++i)
{
rg int edi=read(),type=read();//历史版本edi,询问type
if(type==)
{
rg int kk=read(),x=read();//把kk号的val改成x
root[i]=root[edi];//先把根连过来,再修改!
Modify(root[i],,n,kk,x);//从根开始,左,右端点,修改的编号,修改成的值
}
else
{
rg int kk=read();
printf("%d\n",Query(root[edi],,n,kk));//当前节点(也就是edi时的根),左,右端点,询问第kk号的值
root[i]=root[edi];//这个依题,还是弄过来吧……
}
}
return ;
}
上一篇:java操作es之rest风格基本操作


下一篇:P3919 【模板】可持久化数组 -初步探究主席树