poj 2528 Mayor's posters(数据结构:线段树+离散化)

在一个固定高的展示板上排列海报

海报可以相互覆盖,计算最后可以看到的海报的个数

因为给定的数据量太大了,所以要进行离散化

而普通的离散化并不能解决问题,原因在于这里要表示的是区间

普通的离散化可能会导致某些区间消失,举例如下:

给出下面两个简单的例子应该能体现普通离散化的缺陷:
例子一:1-10 1-4 5-10
例子二:1-10 1-4 6-10
普通离散化后都变成了[1,4][1,2][3,4]
线段2覆盖了[1,2],线段3覆盖了[3,4],那么线段1是否被完全覆盖掉了呢?
例子一是完全被覆盖掉了,而例子二没有被覆盖

为了解决这种缺陷,我们可以在排序后的数组上加些处理,比如说[1,2,6,10]
如果相邻数字间距大于1的话,在其中加上任意一个数字,比如加成[1,2,3,6,7,10],然后再做线段树就好了.

这道题的数组大小一定要特别注意,不知道离散化之后怎么计算数组空间的,因为这个WA了好多次

代码如下:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define MAXN 12010
#define LL long long
using namespace std;

int ans;
int pos[MAXN<<4], tmp[MAXN*3];
int li[MAXN], ri[MAXN];
bool vis[MAXN];

void PushDown(int rt) {
    if(pos[rt] != -1) {
        pos[rt<<1] = pos[rt<<1|1] = pos[rt];
        pos[rt] = -1;
    }
}

int Bin(int key, int n, int tmp[]) {
    int L = 0;
    int R = n;
    while(L <= R) {
        int m = (L+R) >> 1;
        if(tmp[m] == key)
            return m;
        if(tmp[m] > key) {
            R = m-1;
        } else L = m+1;
    }
    return -1;
}

void update(int L, int R, int c, int l, int r, int rt) {
    if(L<=l && r<=R) {
        pos[rt] = c;
        return ;
    }
    PushDown(rt);
    int m = (l+r) >> 1;
    if(L <= m)
        update(L, R, c, l, m, rt<<1);
    if(R > m)
        update(L, R, c, m+1, r, rt<<1|1);
}

void query(int l, int r, int rt) {
    if(pos[rt] != -1) {
        if(!vis[pos[rt]])
            ++ans;
        vis[pos[rt]] = true;
        return ;
    }
    if(l == r)
        return ;
    int m = (l+r) >> 1;
    query(l, m, rt<<1);
    query(m+1, r, rt<<1|1);
}

int main(void) {
    int T, n;
    scanf("%d", &T);
    while(T--) {
        scanf("%d", &n);
        int t = 0;
        for(int i=0; i<n; ++i) {
            scanf("%d%d", &li[i], &ri[i]);
            tmp[t++] = li[i];
            tmp[t++] = ri[i];
        }
        //对tmp进行离散化
        sort(tmp, tmp+t) ;
        int cnt = 1;
        for(int i=1; i<t; ++i) {//滚动数组,节省空间
            if(tmp[i] != tmp[i-1])
                tmp[cnt++] = tmp[i];
        }
        for(int i=cnt-1; i>0; --i) {
            if(tmp[i] != tmp[i-1]+1)
                tmp[cnt++] = tmp[i-1]+1;
        }
        sort(tmp, tmp+cnt);
        
        memset(pos, -1, sizeof(pos));
        for(int i=0; i<n; ++i) {
            int l = Bin(li[i], cnt-1, tmp);
            int r = Bin(ri[i], cnt-1, tmp);
            update(l, r, i, 0, cnt-1, 1);
        }
        ans = 0;
        memset(vis, false, sizeof(vis));
        query(0, cnt-1, 1);
        printf("%d\n", ans);
    }
    return 0;
}