HDU 4417 【线段树+离线处理】

http://acm.hdu.edu.cn/showproblem.php?pid=4417

题意：找出给定区间内，有多少个数小于等于给定的数。用线段树维护的话会超时，要用到线段树的离线操作，对询问与数列都进行从小到大的排序，记录下标。从第一个询问开始，遍历数列，满足小于等于就插入到线段树中相应的位置。答案即为当前线段树中有多少个值。转换成了区间和，记录答案，最后一遍输出。

代码如下：

 #include<stdio.h>
 #include<algorithm>
 const int MAXN = 1e5 + ;
 using namespace std;

 int n, m;
 int ANS[MAXN];

 struct Node
 {
     int id, h;
 }node[MAXN];
 bool cmp1(Node a, Node b)
 {
     return a.h < b.h;
 }

 struct Tree
 {
     int l, r, num;
 }t[MAXN * ];

 struct Query
 {
     int l, r, limit, id;
 }q[MAXN];
 bool cmp2(Query a, Query b)
 {
     return a.limit < b.limit;
 }

 void build(int l, int r, int k)
 {
     t[k].l = l, t[k].r = r, t[k].num = ;
     if(l == r)
         return ;
     int mid = (l + r) / ;
     build(l, mid,  * k);
     build(mid + , r,  * k + );
 }

 void up_date(int id, int k)
 {
     if(t[k].l == t[k].r)
     {
         t[k].num = ;
         return ;
     }
     int mid = (t[k].l + t[k].r) / ;
     if(id <= mid)
         up_date(id,  * k);
     else
         up_date(id,  * k + );
     t[k].num = t[ * k].num + t[ * k + ].num;
 }

 int query(int f_l, int f_r, int k)
 {
     if(f_l <= t[k].l && f_r >= t[k].r)
         return t[k].num;
     int mid = (t[k].l + t[k].r) / ;
     if(f_l > mid)
         return query(f_l, f_r,  * k + );
     else if(f_r <= mid)
         return query(f_l, f_r,  * k);
     else
     {
         return query(f_l, mid,  * k) + query(mid + , f_r,  * k + );
     }
 }

 int main()
 {
     int T, k = ;
     scanf("%d", &T);
     while(T --)
     {
         scanf("%d%d", &n, &m);
         build(, n, );
         for(int i = ; i <= n; i ++)
         {
             scanf("%d", &node[i].h);
             node[i].id = i;
         }
         sort(node + , node +  + n, cmp1);
     //    for(int i = 1; i <= n; i ++)
     //        printf("%d\n", node[i].h);
         printf("Case %d:\n", k ++);
         for(int i = ; i <= m; i ++)
         {
             scanf("%d%d%d", &q[i].l, &q[i].r, &q[i].limit);
             q[i].l ++, q[i].r ++;
             q[i].id = i;
         }
         sort(q + , q +  + m, cmp2);
         int j = ;
         for(int i = ; i <= m; i ++)
         {
             while(node[j].h <= q[i].limit && j <= n)
             {
                 up_date(node[j].id, );
                 j ++;
             }
             ANS[q[i].id] = query(q[i].l, q[i].r, );
         }
         for(int i = ; i <= m; i ++)
             printf("%d\n", ANS[i]);
     }
     return ;
 }