可持久化线段树 (主席树) + 动态修改+查询

题目链接:https://www.luogu.org/problemnew/show/P3919

思路:直接按题意写就行。更新直接更新叶子节点的值就可以了。

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

#define LL long Long
using namespace std;
const int maxn=2e6+10;

int L[maxn*20], R[maxn*20], sum[maxn*20];
int n, q,cnt = 0;
int a[maxn], root[maxn];
#define mid (l+r)/2

int JS(int l, int r)
{
    int rt=++cnt;
    sum[rt]=0;

    if(l==r)
    {
        sum[rt]=a[l];
        return rt;
    }

    L[rt]=JS(l ,mid);
    R[rt]=JS(mid+1, r);

    return rt;
}

int gx(int i, int l, int r, int x, int v)
{
    int rt=++cnt;
    L[rt]=L[i], R[rt]=R[i];//更新就新建节点
    if(l==r)
    {
        sum[rt]=v;
        return rt;
    }
    if(x<=mid)
    {
        L[rt]=gx(L[i], l, mid, x, v);
    }
    else
    {
        R[rt]=gx(R[i], mid+1, r, x, v);
    }
    return rt;
}

int cx(int i, int l, int r, int x)//x数组的下标
{
    if(l==r)
    {
        return sum[i];
    }

    if(x<=mid)
    {
        return cx(L[i], l, mid, x);
    }
    else
    {
        return cx(R[i], mid+1, r, x);
    }

}

int main()
{
    scanf("%d%d", &n, &q);
    for (int i = 1; i <= n; i ++)
    {
        scanf("%d", &a[i]);
    }

    root[0] = JS(1, n);
    for(int i=1;i<=q;i++)
    {
        int w1, w2, w3, w4;
        scanf("%d%d%d",&w1,&w2,&w3);
        if(w2==1)
        {
            scanf("%d",&w4);

            root[i]=gx(root[w1], 1, n, w3, w4);
        }
        else
        {
            printf("%d\n",cx(root[w1], 1, n, w3));
            root[i]=root[w1];
        }
    }

    return 0;
}

全部评论

相关推荐

就是大飞舞:不能发,我暑假找实习的时候就被坑过,把你简历锁了,然后不给你推进度,你还投不了别的部门
点赞 评论 收藏
分享
威猛的小饼干正在背八股:挂到根本不想整理
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务