Luogu-3834 (主席树模板)

代码

//就是一个主席树#include <cstdio>#include <cstring>#define Max 1000000000#define Mid (l+r>>1)#define D(x) (T[T[x2].x].v-T[T[x1].x].v)using namespace std;struct node{    int l,r,v;}T[15000000];int k,n,m,s,t,num,Q,o,st[2000005];void add(int &x1,int x2,int l,int r){    x1=++num;   //x1肯定是一个之前没出现过的点，所以给他分配一个 node     T[x1]=T[x2];    T[x1].v++;    if (l==r) return;    if (l<=o && o<=Mid) {add(T[x1].l,T[x2].l,l,Mid); return;}    add(T[x1].r,T[x2].r,Mid+1,r);}int que(int x1,int x2,int l,int r,int O){    if (l==r) return l;    if (D(l)+O<k) return que(T[x1].r,T[x2].r,Mid+1,r,D(l)+O);    return que(T[x1].l,T[x2].l,l,Mid,O);}int main(){    freopen("1.txt","r",stdin);    scanf("%d%d",&n,&m);    for (int i=1; i<=n; i++){        scanf("%d",&o);        add(st[i],st[i-1],-Max,Max);    }    while (m--){        scanf("%d%d%d",&s,&t,&k);        printf("%d\n",que(st[s-1],st[t],-Max,Max,0));    }}