洛谷 P3919 可持久化线段树 1 之主席树模板(初级)

洛谷P3919题解


传送锚点


摸鱼环节

【模板】可持久化线段树 1(可持久化数组)

题目背景

UPDATE : 最后一个点时间空间已经放大

2021.9.18 增添一组 hack 数据 by @panyf

标题即题意

有了可持久化数组,便可以实现很多衍生的可持久化功能(例如:可持久化并查集)

题目描述

如题,你需要维护这样的一个长度为 $ N $ 的数组,支持如下几种操作

  1. 在某个历史版本上修改某一个位置上的值

  2. 访问某个历史版本上的某一位置的值

此外,每进行一次操作(对于操作2,即为生成一个完全一样的版本,不作任何改动),就会生成一个新的版本。版本编号即为当前操作的编号(从1开始编号,版本0表示初始状态数组)

输入格式

输入的第一行包含两个正整数 $ N, M $, 分别表示数组的长度和操作的个数。

第二行包含$ N $个整数,依次为初始状态下数组各位的值(依次为 $ a_i \(,\) 1 \leq i \leq N $)。

接下来$ M \(行每行包含3或4个整数,代表两种操作之一(\) i $为基于的历史版本号):

  1. 对于操作1,格式为$ v_i \ 1 \ {loc}_i \ {value}i \(,即为在版本\) v_i $的基础上,将 $ a_i} $ 修改为 $ {value}_i $。

  2. 对于操作2,格式为$ v_i \ 2 \ {loc}i \(,即访问版本\) v_i $中的 $ a_i} $的值,注意:生成一样版本的对象应为 \(v_i\)

输出格式

输出包含若干行,依次为每个操作2的结果。

样例 #1

样例输入 #1

5 10
59 46 14 87 41
0 2 1
0 1 1 14
0 1 1 57
0 1 1 88
4 2 4
0 2 5
0 2 4
4 2 1
2 2 2
1 1 5 91

样例输出 #1

59
87
41
87
88
46

提示

数据规模:

对于30%的数据:$ 1 \leq N, M \leq {10}^3 $

对于50%的数据:$ 1 \leq N, M \leq {10}^4 $

对于70%的数据:$ 1 \leq N, M \leq {10}^5 $

对于100%的数据:$ 1 \leq N, M \leq {10}^6, 1 \leq {loc}_i \leq N, 0 \leq v_i < i, -{10}^9 \leq a_i, {value}_i \leq {10}^9$

经测试,正常常数的可持久化数组可以通过,请各位放心

数据略微凶残,请注意常数不要过大

另,此题I/O量较大,如果实在TLE请注意I/O优化

询问生成的版本是指你访问的那个版本的复制

样例说明:

一共11个版本,编号从0-10,依次为:

* 0 : 59 46 14 87 41

* 1 : 59 46 14 87 41

* 2 : 14 46 14 87 41

* 3 : 57 46 14 87 41

* 4 : 88 46 14 87 41

* 5 : 88 46 14 87 41

* 6 : 59 46 14 87 41

* 7 : 59 46 14 87 41

* 8 : 88 46 14 87 41

* 9 : 14 46 14 87 41

* 10 : 59 46 14 87 91


不得不说,发明主席树的HJT可真是个添柴
这道可持久化线段树的板子,今天也是有空干掉了(绝不是之前懒得写)。其小名主席树也是OIer皆知,so what is 主席树?说白了就是完成对某一时刻数据的修改与查询(起码这道题是这样),此时就需要将这些状态全缓存下来,就可以达到目的了。


正片开始

如此持久特别的主席树会不会很难写?The answer is YES,线段树不难写吗? NO,基本实现与线段树无异,只是多了个储存版本的过程。

1.储存信息

  1. 储存树的信息可选择结构体(简单直观)。
  2. 用root数组储存版本信息(初始版本编号为0)。

2.建树部分

与一般建树无异,需要处理出节点编号个数。

code:

void build(int &u,int l,int r)
{
    u=++top;
    if(l==r){tree[u].val=a[l];return;}
    int mid=(l+r)>>1;
    build(tree[u].l,l,mid);
    build(tree[u].r,mid+1,r);
}

3.更新部分

对于每次更新,都需要新建版本节点。

code:

int add(int u)
{
    top++;
    tree[top]=tree[u];//将原先节点信息传递给新建节点
    return top;
}
void update(int &u,int l,int r,int x,int v)
{
    u=add(u);//对于修改部分新建节点
    if(l==r){tree[u].val=v;return;}
    int mid=(l+r)>>1;
    if(x<=mid){update(tree[u].l,l,mid,x,v);}
    else {update(tree[u].r,mid+1,r,x,v);}
}

4.询问部分

这与线段树询问一毛一样。

code:

int query(int u,int l,int r,int x)
{
    if(l==r) return tree[u].val;
    int mid=(l+r)>>1;
    if(x<=mid) return query(tree[u].l,l,mid,x);
    else return query(tree[u].r,mid+1,r,x);
}

5.主体部分

直接见代码。

code:

build(root[0],1,n);//root数组保存版本根节点编号
for(int i=1;i<=m;i++)
{
    int rt,op,x,y;scanf("%d%d%d",&rt,&op,&x);
    if(op==1){scanf("%d",&y);root[i]=root[rt];update(root[i],1,n,x,y);}//操作一,i为版本编号,root[i]储存版本信息
    else{printf("%d\n",query(root[rt],1,n,x));root[i]=root[rt];}//操作二与操作一同理
}

6.注意!!警钟长鸣!!

1.数组大小

由于需要多次新建节点,所以数组大小尽量开大,同时不要忘记tree需要开四倍空间。

2.TLE问题

由于测试点出现毒瘤良心数据,所以在读入与输出时,选择快读快输或是\(scanf\) \(printf\)


完整代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e7-520;
int n,m;
int top,root[N],a[N];
struct node
{
    int l,r,val;
}tree[N*4];
int add(int u)
{
    top++;
    tree[top]=tree[u];
    return top;
}
void build(int &u,int l,int r)
{
    u=++top;
    if(l==r){tree[u].val=a[l];return;}
    int mid=(l+r)>>1;
    build(tree[u].l,l,mid);
    build(tree[u].r,mid+1,r);
}
void update(int &u,int l,int r,int x,int v)
{
    u=add(u);
    if(l==r){tree[u].val=v;return;}
    int mid=(l+r)>>1;
    if(x<=mid){update(tree[u].l,l,mid,x,v);}
    else {update(tree[u].r,mid+1,r,x,v);}
}
int query(int u,int l,int r,int x)
{
    if(l==r) return tree[u].val;
    int mid=(l+r)>>1;
    if(x<=mid) return query(tree[u].l,l,mid,x);
    else return query(tree[u].r,mid+1,r,x);
}
int main()
{
	scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    build(root[0],1,n);//root数组保存版本根节点编号
    for(int i=1;i<=m;i++)
    {
        int rt,op,x,y;scanf("%d%d%d",&rt,&op,&x);
        if(op==1){scanf("%d",&y);root[i]=root[rt];update(root[i],1,n,x,y);}
        else{printf("%d\n",query(root[rt],1,n,x));root[i]=root[rt];}
    }
	return 0;
} 

完结收工!!!!!

个人主页

看完点赞,养成习惯

\(\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\Downarrow\)

posted @ 2024-08-19 10:51  Nightmares_oi  阅读(7)  评论(0编辑  收藏  举报