ZOJ 3299 Fall the Brick (线段树)

本文介绍了一种基于离散化和线段树的数据结构算法,用于解决区间更新与查询的问题。通过预先处理输入数据并使用离散化技巧减少数据范围,结合线段树进行高效区间操作,该算法能有效应对大规模数据集的挑战。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

思路:先离散化,对每个board从高到低排序,依次求每个board对应区间的brick数,求出后对应区间清零。


这题卡内存,AC好不容易啊!!


#include<cstdio>
#include<iostream>
#include<vector>
#include<algorithm>
#include<cstdlib>
#include<map>

#define lson l ,m ,rt<<1
#define rson m+1 ,r ,rt<<1|1
using namespace std;
typedef long long LL;
const int maxn = 100005;

struct Input{
    int id,l,r,h;
    bool operator < (const Input &z) const{
        return h > z.h;
    }
}in[maxn];

int n,q,l[maxn],r[maxn];
int mc[maxn<<2],len;
LL ans[maxn];

int get_id(int x){
    return lower_bound(mc,mc+len,x) - mc;
}

struct SegTree{
    LL sum[1<<20];int c[1<<20];//bool flag[8*maxn];

    void push_up(int rt){
        sum[rt] = sum[rt<<1] + sum[rt<<1|1];
    }
    void push_down(int l,int r,int rt){
        if(c[rt] == 0) return;
        if(l != r){
            c[rt<<1] = (c[rt] != -1) ? c[rt<<1] + c[rt] : -1;
            c[rt<<1|1] = (c[rt] != -1) ? c[rt<<1|1] + c[rt] : -1;
        }
        sum[rt] = (c[rt] != -1) ? sum[rt] + (LL)c[rt]*(mc[r+1]-mc[l]) : 0;
        c[rt] = 0;
    }
    void build(int l,int r,int rt){
        c[rt] = 0;//flag[rt ] = false;
        if(l == r){
            sum[rt] = 0;
            return;
        }
        int m = (l + r) >> 1;
        build(lson);build(rson);
        push_up(rt);
    }
    void update(int a,int b,int x,int l,int r,int rt){
        push_down(l,r,rt);
        if(l > b || r < a) return;
        if(a <= l && r <= b){
            c[rt] = x;
            push_down(l,r,rt);
            return;
        }
        int m = (l + r) >> 1;
        update(a,b,x,lson);update(a,b,x,rson);
        push_up(rt);
    }
    LL query(int a,int b,int l,int r,int rt){
        push_down(l,r,rt);
        if(l > b || r < a) return 0;
        if(a <= l && r <= b) return sum[rt];
        int m = (l + r) >> 1;
        return query(a,b,lson) + query(a,b,rson);
    }
}sol;

int main(){
    while(~scanf("%d%d",&n,&q)){
        int i,o;len = 0;
        for(i = 0; i < n ; i++){
            scanf("%d%d",&l[i],&r[i]);
            mc[len++] = l[i];mc[len++] = r[i];
        }
        for(i = 0; i < q; i++){
            scanf("%d%d%d",&in[i].l,&in[i].r,&in[i].h);in[i].id = i;
            mc[len++] = in[i].l;mc[len++] = in[i].r;
        }
        sort(in,in+q);
        sort(mc,mc+len);
        len = unique(mc,mc+len) - mc;
        o = len;
        sol.build(0,o-1,1);
        for(i = 0;i < n; i++){
            l[i] = get_id(l[i]);r[i] = get_id(r[i]);
            sol.update(l[i],r[i]-1,1,0,o-1,1);
        }
        for(i = 0;i < q; i++){
            in[i].l = get_id(in[i].l);in[i].r = get_id(in[i].r);
            ans[in[i].id] = sol.query(in[i].l,in[i].r-1,0,o-1,1);
            sol.update(in[i].l,in[i].r-1,-1,0,o-1,1);
        }
        for(i = 0; i < q; i++) printf("%lld\n",ans[i]);
        putchar('\n');
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值