主席树

  • msjing on 6.22 17:42:学习笔记还在写
#include<bits/stdc++.h>
#define lson tr[rt].l
#define rson tr[rt].r
using namespace std;
constexpr int maxn=1e6+10,inf=0x7f7f7f7f;
int read()
{
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0' || ch>'9')
    {
        if (ch == '-') f=-1;
        ch=getchar();
    }
    while (ch>='0' && ch<='9')
    {
        x=(x<<1)+(x<<3)+ch-'0';
        ch=getchar();
    }
    return x*f;
}
int n,m,a[maxn];
int root[maxn];
int seg;
struct segtree
{
    struct _ {int l,r,sum;}tr[maxn*100];
    void Pus(int rt) {tr[rt].sum=tr[lson].sum+tr[rson].sum;}
    void Upd(int lrt,int &nrt,int l,int r,int p,int v)
    {
        if (!nrt) nrt=++seg;
        tr[nrt]=tr[lrt];
        if (l == r)
        {
            tr[nrt].sum+=v;
            return;
        }
        int mid=(l+r) >> 1;
        if (p<=mid) Upd(tr[lrt].l,tr[nrt].l=++seg,l,mid,p,v);
        else Upd(tr[lrt].r,tr[nrt].r=++seg,mid+1,r,p,v);
        Pus(nrt);
    }
    int Que(int lrt,int nrt,int l,int r,int p)
    {
        if (l == r) return l;
        int mid=(l+r) >> 1;
        int k=tr[tr[nrt].l].sum-tr[tr[lrt].l].sum;
        if (p<=k) return Que(tr[lrt].l,tr[nrt].l,l,mid,p);
        else return Que(tr[lrt].r,tr[nrt].r,mid+1,r,p-k);
    }
}S;

int main()
{
    n=read(),m=read();for (int i=1;i<=n;i++) a[i]=read(),S.Upd(root[i-1],root[i],0,1e9,a[i],1);
    for (int i=1;i<=m;i++)
    {
        int x=read(),y=read(),z=read();
        printf("%d\n",S.Que(root[x-1],root[y],0,1e9,z));
    }
    return 0;
}
posted @ 2026-07-22 17:43  msjing  阅读(10)  评论(0)    收藏  举报