poj 2528 线段树 lazy Mayor's poster

决定将这个再写一遍，这是以前的代码。

/*
Pro: poj-2528

Sol:

date:2012/07/03
*/
#include<iostream>
#include <cstring>
#include <cstdio>
#include<algorithm>
//这个define还不能用加的，这么低级啊。。。
#define N 10010
using namespace std;
int Case,n,dis[10000100],s[N],e[N],cnt,tp[2 * N],all,tag[N];
struct you{
    int s,e,col;
    you():s(0),e(0),col(0){};//
}lt[N * 8];
void discrete(){
    all = 0;
    for(int i = 0;  i < n; i ++){
        if(!dis[s[i]])
            {tp[all++] = s[i];   dis[s[i]] = 1;}//去重！！！
        if(!dis[e[i]]){
            tp[all ++] = e[i];   dis[e[i]] = 1;
        }
    }
    sort(tp,tp + all);

    for(int i = 0; i < all; i ++){//把排好序的端点依次映射到1,2,...,all
        dis[tp[i]] = i + 1;//离散化肿么用???,颜色从1开始
    }
}
void creat(int sp,int ep,int p){//如何调用creat???
    lt[p].s = sp;//
    lt[p].e = ep;

    if(sp == ep) return;
    int mid = (sp + ep) >> 1;

    creat(sp,mid, 2 * p);
    creat(mid + 1, ep, 2 * p + 1);      return ;
}
void insert(int a,int b, int p, int color){//p 是干什么的?
//[a,b]与[s,e]完全无交集
//说明[a,b]不被[s,e]所在的子树包含，无需向下搜索
    if(b < lt[p].s || a > lt[p].e) return ;
//完全覆盖,  [a,b]完全覆盖[s,e]
    if(a <= lt[p].s && b >= lt[p].e) {
        lt[p].col = color;  return ;
        //[s,e]染单色，暂时无需对[s,e]的子树操作(这是由线段树的搜索机制决定的)
    }
//[a,b]颜色部分覆盖[s,e]，本身无色或单色,需对子树操作
    if(lt[p].col >= 0){//单色或者无色  0 --- 无色   1 ---- 单色   -1 ----多种颜色
        lt[p * 2].col = lt[p * 2 + 1].col = lt[p].col; //给孩子保存
        lt[p].col = -1;   //多色
    }
//本身多色，从左子树和右子树开始搜
    insert(a,b,p * 2,color);       //觉得可以改进
    insert(a,b,p * 2 + 1, color);

    return;
}
void dfs(int p){//这里p又是干什么的---p是一个下标
//无色
    if(lt[p].col == 0) return;
//单色
    if(lt[p].col > 0){
        if(!tag[lt[p].col]){//数颜色
            tag[lt[p].col] = 1;
            cnt ++;
        }
        return ; //单色数了颜色之后就返回
    }
//多色
        dfs(p * 2);  dfs(p * 2 + 1);    return ;

}
void solution(){
    cnt = 0;    memset(tag,0,sizeof(tag));

    discrete();
    creat(1,all,1);//从1开始建树

    for(int i = 0; i < n; i ++){//插入
        insert(dis[s[i]],dis[e[i]], 1, i + 1);    //离散化的使用
    }
    dfs(1);//数颜色
    printf("%d\n",cnt);
    return ;
}
int main(){
    cin >> Case;
    while(Case --){
        cin >> n;   memset(dis,0,sizeof(dis));
        for(int i = 0 ; i < n; i ++){
            scanf("%d%d",s + i, e + i);
        }
        solution();
    }
    return 0;
}

最新代码：（-1表示这个区间多色）

/*
Pro: poj-2528

Sol:

date:2012/09/15
*/
#include<iostream>
#include <cstring>
#include <cstdio>
#include<algorithm>
#define lson l,m, rt << 1
#define rson m + 1, r, rt << 1|1
#define havem int m = (l + r) >> 1
//这个define还不能用加的，这么低级啊。。。
const int maxn = 10010;
using namespace std;
int T, col[maxn << 4],n,li[maxn],ri[maxn],X[maxn * 3],cnt;
bool hash[maxn  << 1];
void push_dn(int rt){
    if(col[rt] != -1){
        col[rt << 1] = col[rt << 1 | 1] = col[rt];
        col[rt] = -1;
    }
}
void push_up(int rt){
    if(col[rt << 1] != col[rt << 1 | 1]) col[rt] = -1;
}
void update(int L, int R, int cc,int l, int r, int rt){
    if(L <= l && r <= R){
        col[rt] = cc; return ;
    }havem;
    push_dn(rt);
    if(L <= m) update(L,R,cc,lson);
    if(R > m) update(L,R,cc,rson);
//    push_up(rt);
}
void query(int L, int R, int l , int r, int rt){
    if(L <= l && r <= R){
        if(col[rt] != -1){
            if(!hash[col[rt]]) cnt ++;
            hash[col[rt]] = true;
            return ;
        }
    }havem;
    if(l == r) return ;
    push_dn(rt);
    if(L <= m) query(L,R,lson);
    if(R > m) query(L,R,rson);
}
int main(){
    scanf("%d",&T);
    while(T --){
        scanf("%d",&n);
        int xsub = 0;
        for(int i = 0; i < n; i ++){
            scanf("%d%d",&li[i],&ri[i]);
            X[xsub ++] = li[i];
            X[xsub ++] = ri[i];
        }
        sort(X, X + xsub);//下标这里得特别注意
        int m = 1;
        for(int i = 1; i < xsub; i ++){
            if(X[i] != X[i - 1]) X[m ++] = X[i];
        }
        xsub = m;
        for(int i = 1; i < xsub; i ++){
            if(X[i] != X[i - 1] + 1) X[m ++] = X[i - 1] + 1;
        }
        sort(X,X + m);
        memset(col, - 1, sizeof(col)); //-1 即是开始，又是表示多色
        for(int i = 0; i < n; i ++){
            int l = lower_bound(X,X + m,li[i]) - X + 1;
            int r = lower_bound(X,X + m,ri[i]) - X + 1;
            update(l,r,i,1,m,1);
        }
        memset(hash,0,sizeof(hash));
        cnt = 0;
        query(1,m,1,m,1);
//        int cnt = 0;
//        for(int i = 0; i < n; i ++)//这里必须是n，m不一定比n大
//            if(hash[i])
//                cnt ++;
        printf("%d\n",cnt);
    }
    return 0;
}