ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

[P3834]【模板】可持久化线段树 2(主席树)

2020-09-13 09:02:18  阅读:230  来源: 互联网

标签:src ch int 线段 modify P3834 key ind 模板


复习一下主席树板子

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

#define int long long
const int N = 10000005;
const int V = 1e9;
int n,m,src[N],a[N],ch[N][2],root[N],ind;

void copynode(int dest,int src)
{
    ch[dest][0]=ch[src][0];
    ch[dest][1]=ch[src][1];
    a[dest]=a[src];
}

void modify(int p,int l,int r,int pos,int key)
{
    a[p]+=key;
    if(l<r)
    {
        if(pos<=((l+r)>>1))
        {
            copynode(++ind,ch[p][0]);
            ch[p][0]=ind;
            modify(ch[p][0],l,(l+r)>>1,pos,key);
        }
        else
        {
            copynode(++ind,ch[p][1]);
            ch[p][1]=ind;
            modify(ch[p][1],((l+r)>>1)+1,r,pos,key);
        }
    }
}

void modify(int ver,int pos,int key)
{
    copynode(++ind,root[ver-1]);
    root[ver]=ind;
    modify(root[ver],-V,V,pos,key);
}

int query(int p,int q,int l,int r,int k)
{
    if(l==r)
    {
        return l;
    }
    else
    {
        if(k>a[ch[p][0]]-a[ch[q][0]])
        {
            return query(ch[p][1],ch[q][1],((l+r)>>1)+1,r,k-a[ch[p][0]]+a[ch[q][0]]);
        }
        else
        {
            return query(ch[p][0],ch[q][0],l,(l+r)>>1,k);
        }
    }
}

int query(int verl,int verr,int k)
{
    return query(root[verr],root[verl],-V,V,k);
}

signed main()
{
    ios::sync_with_stdio(false);

    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>src[i];
        modify(i,src[i],1);
    }
    a[0]=0;ch[0][0]=ch[0][1]=0;
    for(int i=1;i<=m;i++)
    {
        int l,r,k;
        cin>>l>>r>>k;
        cout<<query(l-1,r,k)<<endl;
    }
}

标签:src,ch,int,线段,modify,P3834,key,ind,模板
来源: https://www.cnblogs.com/mollnn/p/13660066.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有