HDU 4417 Super Mario

线段树的离线查询,把查询存起来按照h排个序,把砖的高度也存起来按照h排个序,然后按需把砖头update进线段树就好了。

一开始傻了吧唧的h每增加一次rebuild一次树,后来才意识到build一次是nlogn的,我囧,还傻傻的搞了一些常数优化。。

最后看着改了一下,写得比较挫。。不过速度还是比较快的

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

#define M ((L+R)>>1)
#define lson rt<<1,L,M
#define rson rt<<1|1,M+1,R
#define root 1,1,N

const int maxn = 100001;

struct q {
    int l,r,h,res;
};

struct brick {
    int val,pos;
};

int sum[maxn << 2],ql,qr,tar;
q querys[maxn],*pquerys[maxn];
brick b[maxn];
char tmp[15];

bool cmp(q *a,q *b) {
    return (a->h) < (b->h);
}

bool cmp1(brick a,brick b) {
    return a.val < b.val;
}

inline void pushup(int rt) {
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}

void update(int rt,int L,int R) {
    if(L == R) sum[rt] = 1;
    else {
        if(tar <= M) update(lson);
        else update(rson);
        pushup(rt);
    }
}

int query(int rt,int L,int R) {
    if(ql <= L && qr >= R) return sum[rt];
    else {
        int ans = 0;
        if(ql <= M) ans += query(lson);
        if(qr > M) ans += query(rson);
        return ans;
    }
}

int getnum() {
    int ans = 0,i = 1;
    while((tmp[0] = getchar()) == ' ' || tmp[0] == '
');
    while((tmp[i] = getchar()) != ' ' && tmp[i] != '
') i++;
    for(int k = 0;k < i;k++) ans = ans * 10 + tmp[k] - '0';
    return ans;
}

void putnum(int num) {
    int i = 0;
    tmp[0] = '0';
    while(num) {
        tmp[i++] = num % 10 + '0';
        num /= 10;
    }
    i--;
    if(i < 0) puts("0");
    else {
        while(i >= 0) putchar(tmp[i--]);
        putchar('
');
    }
}

int main() {
    int T,N,m,now;
    T = getnum();
    for(int k = 1;k <= T;k++) {
        memset(sum,0,sizeof(sum));
        int bpos = 0,hh = -1;
        N = getnum(); m = getnum();
        for(int i = 0;i < N;i++) {
            b[i].val = getnum();
            b[i].pos = i + 1;
        }
        for(int i = 0;i < m;i++) {
            querys[i].l = getnum();
            querys[i].r = getnum();
            querys[i].h = getnum();
            pquerys[i] = querys + i;
        }
        sort(pquerys,pquerys + m,cmp);
        sort(b,b + N,cmp1);
        for(int i = 0;i < m;i++) {
            ql = pquerys[i]->l + 1; qr = pquerys[i]->r + 1;
            if(hh < pquerys[i]->h) {
                hh = pquerys[i]->h;
                while(b[bpos].val <= hh && bpos < N) {
                    tar = b[bpos++].pos;
                    update(root);
                }
            }
            pquerys[i]->res = query(root);
        }
        printf("Case %d:
",k);
        for(int i = 0;i < m;i++) putnum(querys[i].res);
    }
    return 0;
}
原文地址:https://www.cnblogs.com/rolight/p/3468390.html