[主席树]HDOJ4417 Super Mario

题意：n个数 m个询问  ($n、m \le 10^5$)

每个询问有l, r, k  问的是[l, r]区间内有多少个数小于等于k

用主席树做的话查询第i小的数与k比较即可

 #define lson l, m
 #define rson m+1, r
 const int N=1e5+;
 int L[N<<], R[N<<], sum[N<<];
 int tot;
 int a[N], T[N], Hash[N];
 int build(int l, int r)
 {
     int rt=(++tot);
     sum[rt]=;
     if(l<r)
     {
         int m=(l+r)>>;
         L[rt]=build(lson);
         R[rt]=build(rson);
     }
     return rt;
 }

 int update(int pre, int l, int r, int x)
 {
     int rt=(++tot);
     L[rt]=L[pre], R[rt]=R[pre], sum[rt]=sum[pre]+;
     if(l<r)
     {
         int m=(l+r)>>;
         if(x<=m)
             L[rt]=update(L[pre], lson, x);
         else
             R[rt]=update(R[pre], rson, x);
     }
     return rt;
 }

 int query(int u, int v, int l, int r, int k)
 {
     if(l>=r)
         return l;
     int m=(l+r)>>;
     int num=sum[L[v]]-sum[L[u]];
     if(num>=k)
         return query(L[u], L[v], lson, k);
     else
         return query(R[u], R[v], rson, k-num);
 }

 int main()
 {
     int t, ca=;
     scanf("%d", &t);
     while(t--)
     {
         tot=;
         int n, m;
         scanf("%d%d", &n, &m);
         for(int i=; i<=n; i++)
         {
             scanf("%d", &a[i]);
             Hash[i]=a[i];
         }
         sort(Hash+, Hash+n+);
         int d=unique(Hash+, Hash+n+)-Hash-;
         T[]=build(, d);
         for(int i=; i<=n; i++)
         {
             int x=lower_bound(Hash+, Hash+d+, a[i])-Hash;
             T[i]=update(T[i-], , d, x);
         }
         printf("Case %d:\n", ca++);
         while(m--)
         {
             int l, r, k;
             scanf("%d%d%d", &l, &r, &k);
             l++, r++;
             int ll=, rr=r-l+;
             int ans=;
             while(ll<=rr)
             {
                 int mm=(ll+rr)>>;
                 int tmp=Hash[query(T[l-], T[r], , d, mm)];
                 if(tmp<=k)
                 {
                     if(mm==r-l+ || Hash[query(T[l-], T[r], , d, mm+)]>k)
                     {
                         ans=mm;
                         break;
                     }
                     ll=mm+;
                 }
                 else
                 {
                     if(mm== || Hash[query(T[l-], T[r], , d, mm-)]<=k)
                     {
                         ans=mm-;
                         break;
                     }
                     rr=mm-;
                 }
             }
             printf("%d\n", ans);
         }
     }
     return ;
 }

HDOJ 4417