POJ - 2528 Mayor's posters(线段树＋离散化)

题目大意：有N张海报，现在按顺序将这些海报贴在墙上，每张海报占的[l,r]，问最后墙上能看到几张海报

解题思路：线段树的区间修改，但是这题数据有点大，所以要离散化下
接着就是进行区间修改了，但是这题有个坑点，如果直接离散话的化，有可能会错的
假设有三组数据，[1,10], [1,4] [6,10]
离散化后，1是1，4是2，6是3，10是4
贴第一张海报时：墙的1~4被染为1；
第二张海报时：墙的1~2被染为2，3~4仍为1；
第三张海报时：墙的3~4被染为3，1~2仍为2。
正常离散的话，应该是要输出2的，因为第一张海报被第二张和第三张海报覆盖了，但实际情况并不是，墙4－5间还能看到第1张海报，所以答案应该是3
所以，我们要特殊处理一下离散化，如果两个数之间的差大于1的话，那么我就多离散一个，这样就可以保证距离大于1的，不会被意外覆盖了

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 20010;
int left[N], right[N], num[N << 2], setv[N << 3];
int n, q, ans;
bool vis[N << 2];

void init() {
    scanf("%d", &q);
    int cnt = 0;
    for (int i = 0; i < q; i++) {
        scanf("%d%d", &left[i], &right[i]);
        num[++cnt] = left[i];
        num[++cnt] = right[i];
    }

    sort(num + 1, num + cnt + 1);
    n = 1;
    for (int i = 2; i <= cnt; i++)
        if (num[i] != num[i - 1]) num[++n] = num[i];

    for (int i = cnt; i > 1; i--) 
        if (num[i] - num[i - 1] > 1) num[++n] = num[i] - 1;

    sort(num + 1, num + n + 1);
    memset(setv, -1, sizeof(setv));
}

void PushDown(int u) {
    if (setv[u] == -1) return ;
    setv[u << 1] = setv[(u << 1) + 1] = setv[u];
    setv[u] = -1;
}

void modify(int u, int l, int r, int L, int R, int c) {
    if (L <= l && r <= R) {
        setv[u] = c;
        return;
    }
    PushDown(u);
    int mid = (l + r) >> 1;
    if (L <= mid) modify(u << 1, l, mid, L, R, c);
    if (R > mid) modify((u << 1) + 1, mid + 1, r, L, R, c);
}

int find(int val) {
    int l = 1, r = n;
    while (l <= r) {
        int mid = (l + r) >> 1;
        if (num[mid] == val) return mid;
        else if (num[mid] < val) l = mid + 1;
        else r = mid - 1;
    }
}

void Query(int u, int l, int r) {
    if (setv[u] != -1 || l == r) {
        if (l == r && setv[u] == -1) return; 
        if (!vis[setv[u]]) {
            vis[setv[u]] = true;
            ans++;
        }
        return ;
    }
    int mid = (l + r) >> 1;
    Query(u << 1, l, mid);
    Query((u << 1) + 1, mid + 1, r);
}

void solve() {
    for (int i = 0; i < q; i++) {
        int l = find(left[i]);
        int r = find(right[i]);
        modify(1, 1, n, l, r, i + 1);
    }
    memset(vis, 0, sizeof(vis));
    ans = 0;
    Query(1, 1, n);
    printf("%d\n", ans);
}

int main() {
    int test;
    scanf("%d", &test);
    while (test--) {
        init();
        solve();
    }
    return 0;
}