poj2528(线段树区间替换&离散化)

题目链接: http://poj.org/problem?id=2528

题意: 第一行输入一个 t 表 t 组输入, 对于每组输入: 第一行  n 表接下来有 n 行形如 l, r 的输入, 表在区间 [l, r] 贴一张海报, 问最终能看见几张不同的海报；

思路: 线段树区间替换, 每次 update 时都会给对应区间加一个 lazy 标记, 最终统计标记的种数即为答案；

注意: 题目给出的 l, r 很大, 需要离散化, 和普通的离散化不同, 因为本题中每个单位是代表长度为一的一个区间,而非一个点, 所以对于两个数 a, b, 若 b > a + 1, 

hash[a] = cnt, 那么hash[b] = cnt + 2 而非 hahs[b] = cnt + 1；不然对于一些数据会出错.

代码:

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <map>
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1
using namespace std;

const int MAXN = 1e4 + 10;
int l[MAXN], r[MAXN], gel[MAXN << 1], lazy[MAXN << 4];
bool vis[MAXN << 1];
map<int, int> hash;
int ans;

void push_down(int rt){//将当前标记向下一层更新
    if(lazy[rt] != -1){
        lazy[rt << 1] = lazy[rt << 1 | 1] = lazy[rt];
        lazy[rt] = -1;
    }
}

void update(int L, int R, int x, int l, int r, int rt){//区间替换
    if(L <= l && R >= r){
        lazy[rt] = x;
        return;
    }
    push_down(rt);
    int mid = (l + r) >> 1;
    if(L <= mid) update(L, R, x, lson);
    if(R > mid) update(L, R, x, rson);
}

void query(int l, int r, int rt){
    if(lazy[rt] != -1){//标记的种数即为答案
        if(!vis[lazy[rt]]){
            ans++;
            vis[lazy[rt]] = true;
        }
        return;
    }
    if(l == r) return;
    int mid = (l + r) >> 1;
    query(lson);
    query(rson);
}

int main(void){
    int t, n;
    scanf("%d", &t);
    while(t--){
        int indx = 0, cnt = 1;
        scanf("%d", &n);
        for(int i = 0; i < n; i++){
            scanf("%d%d", &l[i], &r[i]);
            gel[indx++] = l[i];
            gel[indx++] = r[i];
        }
        sort(gel, gel + indx);
        hash[gel[0]] = cnt;
        for(int i = 1; i < indx; i++){ //hash
            if(gel[i] > gel[i - 1] + 1){
                cnt += 2;
                hash[gel[i]] = cnt;
            }else if(gel[i] > gel[i - 1]){
                cnt += 1;
                hash[gel[i]] = cnt;
            }
        }
        memset(lazy, -1, sizeof(lazy));
        memset(vis, false, sizeof(vis));
        for(int i = 0; i < n; i++){
            update(hash[l[i]], hash[r[i]], i, 1, cnt, 1);
        }
        ans = 0;
        query(1, cnt, 1);
        printf("%d
", ans);
    }
    return 0;
}