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

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

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

代码如下：

#include<stdio.h>
#include<algorithm>
const int MAXN = 1e5 + 10;
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 * 4];

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 = 0;
    if(l == r)
        return ;
    int mid = (l + r) / 2;
    build(l, mid, 2 * k);
    build(mid + 1, r, 2 * k + 1);
}

void up_date(int id, int k)
{
    if(t[k].l == t[k].r)
    {
        t[k].num = 1;
        return ;
    }
    int mid = (t[k].l + t[k].r) / 2;
    if(id <= mid)
        up_date(id, 2 * k);
    else
        up_date(id, 2 * k + 1);
    t[k].num = t[2 * k].num + t[2 * k + 1].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) / 2;
    if(f_l > mid)
        return query(f_l, f_r, 2 * k + 1);
    else if(f_r <= mid)
        return query(f_l, f_r, 2 * k);
    else
    {
        return query(f_l, mid, 2 * k) + query(mid + 1, f_r, 2 * k + 1);
    }
}

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