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] == '\n');
    while((tmp[i] = getchar()) != ' ' && tmp[i] != '\n') 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('\n');
    }
}

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:\n",k);
        for(int i = 0;i < m;i++) putnum(querys[i].res);
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值