HDU 4417 Super Mario（树状数组 离线处理）

- HDU 4417 -

Super Mario

Time Limit: 2000/1000 MS (Java/Others) | Memory Limit: 32768/32768 K (Java/Others)

题意：

情境：超级马里奥要去救可怜的公主殿下，路上每隔一个单位长度就有一个砖块，求他在某段路上能击中多少砖块。
第一行給定一个整数T，表示测试案例数；
每组测试案例第一行給定两个整数 n、m，表示路的长度和询问的次数；
接下来一行有 n 个整数 h_i，表示每个砖的高度；
接下来 m 行，每行有 3 个整数 L，R，H，[L, R] 是查询区间，H 是马里奥能跳的最大高度；
对于每组测试，第一行输出： "Case X: "（X 表示从 1 开始的第几组测试案例）；第二行输出马里奥能击中的砖块数。

数据范围：

1 <= n <=10⁵， 1 <= m <= 10⁵，0<=h<=1000000000，0 <= L <= R < n， 0 <= H <= 1000000000

解题思路：

树状数组 离线处理

这题 h 的数据范围太大，不能直接开那么大的数组，所以可以用结构体来存储输入数据的顺序和大小，对每次询问采用离线处理，也用结构体存储其顺序、区间和大小，然后对两个结构体从小到大排序，这样一来只需要扫描一遍这两组数据就能得到答案了。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <stack>
#include <queue>
#include <vector>
#include <map>
#include <set>
using namespace std;
#define INF 0x3f3f3f
#define zero 1e-7

typedef long long ll;
const int N=1e5+5;

int n, m, sum[N], ans[N];
struct node1 {
    int h, id;
}br[N];

struct node2 {
    int l, r, id, h;
}q[N];

bool cmp1(node1 x, node1 y) {
    return x.h<y.h;
}

bool cmp2(node2 x, node2 y) {
    return x.h<y.h;
}

int lowbit(int x) {
    return x&(-x);
}

void add(int x, int c) {//x要从1开始，不能是0，要不然没法儿求lowbit
    while(x<=n) {
        sum[x]+=c;
        x+=lowbit(x);
    }
    return ;
}

int query(int x) {
    int res=0;
    while(x) {
        res+=sum[x];
        x-=lowbit(x);
    }
    return res;
}

int main() {
    int T;
    scanf("%d", &T);
    for(int t=1; t<=T; t++) {
        memset(sum, 0, sizeof(sum));
        memset(ans, 0, sizeof(ans));
        scanf("%d %d", &n, &m);
        for(int i=0; i<n; i++) {
            scanf("%d", &br[i].h);
            br[i].id=i+1;
        }
        for(int i=0; i<m; i++) {
            scanf("%d %d %d", &q[i].l, &q[i].r, &q[i].h);
            q[i].id=i;
        }
        sort(br, br+n, cmp1);
        sort(q, q+m, cmp2);
        int i=0, j=0;
        while(i<m) {
            while(j<n) {
                if(br[j].h>q[i].h) break;
                add(br[j].id, 1);
                j++;
            }
            while(i<m) {
                if(j<n && br[j].h<=q[i].h) break;
                ans[q[i].id]=query(q[i].r+1)-query(q[i].l);
                i++;
            }
        }
        printf("Case %d:\n", t);
        for(int i=0; i<m; i++)
            printf("%d\n", ans[i]);
    }
    return 0;
}