主席树

主席树

区间第k大的值

#include <vector>
#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
#define ll long long
const int maxn = 1e5+6;
int n,m,cnt,root[maxn],a[maxn],x,y,k;
struct node{int l,r,sum;}T[maxn*40];
vector<int>v;
int getid(int x){return lower_bound(v.begin(),v.end(),x)-v.begin()+1;}
void update(int l,int r,int &x,int y,int pos){
    T[++cnt]=T[y],T[cnt].sum++,x=cnt;
    if(l==r)return;
    int mid=(l+r)/2;
    if(mid>=pos)update(l,mid,T[x].l,T[y].l,pos);
    else update(mid+1,r,T[x].r,T[y].r,pos);
}
int query(int l,int r,int x, int y,int k)
{
    if(l==r)return l;
    int mid=(l+r)/2;
    int sum=T[T[y].l].sum-T[T[x].l].sum;
    if(sum>=k)return query(l,mid,T[x].l,T[y].l,k);
    else return query(mid+1,r,T[x].r,T[y].r,k-sum);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]),v.push_back(a[i]);
    sort(v.begin(),v.end()),v.erase(unique(v.begin(),v.end()),v.end());
    for(int i=1;i<=n;i++)update(1,n,root[i],root[i-1],getid(a[i]));
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&x,&y,&k),printf("%d\n",v[query(1,n,root[x-1],root[y],k)-1]);
    return 0;
}

icpc昆明 Stone Games

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 1e6+6;

int n,m,cnt,root[maxn],a[maxn],x,y,k;
struct node{int l,r;ll sum;}T[maxn*40];
vector<int>v;
int len;
int getid(int x){return lower_bound(v.begin(),v.end(),x)-v.begin()+1;}
int get(ll x)
{
    if(v[len-1]<=x) return len;
    int pos=upper_bound(v.begin(),v.end(),x)-v.begin()+1;
    pos--;
    return pos;
}
void update(int l,int r,int &x,int y,int pos,int val){
    T[++cnt]=T[y],T[cnt].sum+=val,x=cnt;
    if(l==r)return;
    int mid=(l+r)/2;
    if(mid>=pos)update(l,mid,T[x].l,T[y].l,pos,val);
    else update(mid+1,r,T[x].r,T[y].r,pos,val);
}
ll query(int l,int r,int x, int y,int k)
{
    if(r<=k)return T[y].sum-T[x].sum;
    ll ans=0;
    int mid=(l+r)>>1;
    if(l<=k)ans+=query(l,mid,T[x].l,T[y].l,k);
    if(mid+1<=k)ans+=query(mid+1,r,T[x].r,T[y].r,k);
    return ans;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]),v.push_back(a[i]);
    sort(v.begin(),v.end()),v.erase(unique(v.begin(),v.end()),v.end());
    for(int i=1;i<=n;i++)update(1,n,root[i],root[i-1],getid(a[i]),a[i]);
    ll last=0;len=v.size();
    while(m--){
        int l,r;
        scanf("%d%d",&l,&r);
        l=(l+last)%n+1;
        r=(r+last)%n+1;
        if(l>r)swap(l,r);
        ll S=0;
        while(1){
            int tmp=get(S+1);
            if(tmp==0) break;
            ll SS=query(1,n,root[l-1],root[r],tmp);
            if(SS==S)break;
            S=SS;
        }
        last=S+1;
        printf("%lld\n",last);
    }
    return 0;
}