ZOJ 3299 Fall the Brick (线段树)

最新推荐文章于 2016-07-09 09:03:10 发布

Neutralzz

最新推荐文章于 2016-07-09 09:03:10 发布

阅读量284

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构

本文链接：https://blog.youkuaiyun.com/qq_26572969/article/details/47396971

数据结构专栏收录该内容

21 篇文章

订阅专栏

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

摘要生成于 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;
}