[主席树]HDOJ4417 Super Mario

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

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

 

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

 

  1 #define lson l, m

  2 #define rson m+1, r

  3 const int N=1e5+5;

  4 int L[N<<5], R[N<<5], sum[N<<5];

  5 int tot;

  6 int a[N], T[N], Hash[N];

  7 int build(int l, int r)

  8 {

  9     int rt=(++tot);

 10     sum[rt]=0;

 11     if(l<r)

 12     {

 13         int m=(l+r)>>1;

 14         L[rt]=build(lson);

 15         R[rt]=build(rson);

 16     }

 17     return rt;

 18 }

 19 

 20 int update(int pre, int l, int r, int x)

 21 {

 22     int rt=(++tot);

 23     L[rt]=L[pre], R[rt]=R[pre], sum[rt]=sum[pre]+1;

 24     if(l<r)

 25     {

 26         int m=(l+r)>>1;

 27         if(x<=m)

 28             L[rt]=update(L[pre], lson, x);

 29         else

 30             R[rt]=update(R[pre], rson, x);

 31     }

 32     return rt;

 33 }

 34 

 35 int query(int u, int v, int l, int r, int k)

 36 {

 37     if(l>=r)

 38         return l;

 39     int m=(l+r)>>1;

 40     int num=sum[L[v]]-sum[L[u]];

 41     if(num>=k)

 42         return query(L[u], L[v], lson, k);

 43     else

 44         return query(R[u], R[v], rson, k-num);

 45 }

 46 

 47 int main()

 48 {

 49     int t, ca=1;

 50     scanf("%d", &t);

 51     while(t--)

 52     {

 53         tot=0;

 54         int n, m;

 55         scanf("%d%d", &n, &m);

 56         for(int i=1; i<=n; i++)

 57         {

 58             scanf("%d", &a[i]);

 59             Hash[i]=a[i];

 60         }

 61         sort(Hash+1, Hash+n+1);

 62         int d=unique(Hash+1, Hash+n+1)-Hash-1;

 63         T[0]=build(1, d);

 64         for(int i=1; i<=n; i++)

 65         {

 66             int x=lower_bound(Hash+1, Hash+d+1, a[i])-Hash;

 67             T[i]=update(T[i-1], 1, d, x);

 68         }

 69         printf("Case %d:\n", ca++);

 70         while(m--)

 71         {

 72             int l, r, k;

 73             scanf("%d%d%d", &l, &r, &k);

 74             l++, r++;

 75             int ll=1, rr=r-l+1;

 76             int ans=0;

 77             while(ll<=rr)

 78             {

 79                 int mm=(ll+rr)>>1;

 80                 int tmp=Hash[query(T[l-1], T[r], 1, d, mm)];

 81                 if(tmp<=k)

 82                 {

 83                     if(mm==r-l+1 || Hash[query(T[l-1], T[r], 1, d, mm+1)]>k)

 84                     {

 85                         ans=mm;

 86                         break;

 87                     }

 88                     ll=mm+1;

 89                 }

 90                 else

 91                 {

 92                     if(mm==1 || Hash[query(T[l-1], T[r], 1, d, mm-1)]<=k)

 93                     {

 94                         ans=mm-1;

 95                         break;

 96                     }

 97                     rr=mm-1;

 98                 }

 99             }

100             printf("%d\n", ans);

101         }

102     }

103     return 0;

104 }

HDOJ 4417