hdu_4417_Super Mario(主席树)

题目链接：hdu_4417_Super Mario

题意：

给你n个树，有m个询问，每个询问有一个区间和一个k,问你这个区间内不大于k的数有多少个。

题解:

考虑用主席树的话就比较裸，当然也可以用其他的写

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=a;i<=b;i++)
using namespace std;

const int N=1e5+7;
int a[N],t,n,m,hsh[N],hsh_len,root[N],tot,x,y,pos,ic=1,k;
struct node{int l,r,sum;}T[N*40];

inline int getid(int x){return lower_bound(hsh+1,hsh+1+hsh_len,x)-hsh;}
inline int getd(int x){return upper_bound(hsh+1,hsh+1+hsh_len,x)-hsh-1;}

void update(int &x,int y,int pos,int l=1,int r=n)
{
    T[++tot]=T[y],T[tot].sum++,x=tot;
    if(l==r)return;
    int m=(l+r)>>1;
    if(pos<=m)update(T[x].l,T[y].l,pos,l,m);
    else update(T[x].r,T[y].r,pos,m+1,r);
}

int query(int x,int y,int pos,int l=1,int r=n)
{
    if(l==r)return T[y].sum-T[x].sum;
    int m=(l+r)>>1;
    if(pos<=m)return query(T[x].l,T[y].l,pos,l,m);
    else return T[T[y].l].sum-T[T[x].l].sum+query(T[x].r,T[y].r,pos,m+1,r);
}

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        printf("Case %d:
",ic++);
        scanf("%d%d",&n,&m),tot=0;
        F(i,1,n)scanf("%d",a+i),hsh[i]=a[i];
        sort(hsh+1,hsh+1+n),hsh_len=unique(hsh+1,hsh+1+n)-hsh-1;
        F(i,1,n)update(root[i],root[i-1],getid(a[i]));
        F(i,1,m)scanf("%d%d%d",&x,&y,&pos),printf("%d
",(k=getd(pos))==0?0:query(root[x],root[y+1],k));
    }
    return 0;
}