Super Mario主席树查询区间排名

Super Mario

 

给你一个数列（长度n），m次询问，每次询问内有多少个数比小

 

input

t

n m

output

嘛，大概就这么回事了，考虑主席树，在查询k的的基础上改掉query即可

 

 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e5+7;
int t,n,m,a[maxn],root[maxn],cnt,x,y,k;
vector<int> v;
int getid(int x){return lower_bound(v.begin(),v.end(),x)-v.begin()+1;}
struct node{int l,r,sum;}T[maxn*40];
void update(int l,int r,int &x,int y,int k){
    T[++cnt] = T[y],T[cnt].sum++,x = cnt;
    if(l == r)return;
    int mid = (l+r)/2;
    if(k<=mid)update(l,mid,T[x].l,T[y].l,k);
    else update(mid+1,r,T[x].r,T[y].r,k);
}
int query(int l,int r,int x,int y,int k){
    if(r<=k)return T[y].sum-T[x].sum;
    int mid = (l+r)/2;
    if(k<=mid)return query(l,mid,T[x].l,T[y].l,k);
    return T[T[y].l].sum-T[T[x].l].sum+query(mid+1,r,T[x].r,T[y].r,k);
}
int main(){
    int Case = 0;
    for(scanf("%d",&t);t--;){
        v.clear(),cnt = 0;
        printf("Case %d:\n",++Case);
        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]));
        while(m--){
            scanf("%d%d%d",&x,&y,&k);
            x++,y++;
            k = upper_bound(v.begin(),v.end(),k)-v.begin();
            if(k)printf("%d\n",query(1,n,root[x-1],root[y],k));
            else printf("0\n");
        }
    }
    return 0;
}