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;
}