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 + ;
 int l[MAXN], r[MAXN], gel[MAXN << ], lazy[MAXN << ];
 bool vis[MAXN << ];
 map<int, int> hash;
 int ans;

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

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

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