POJ 2528 区间染色，求染色数目，离散化

  1 #include <cstdio>
  2 #include <cstring>
  3 #include <algorithm>
  4 #include <iostream>
  5 #include <vector>
  6 #include <queue>
  7 #include <cmath>
  8 #include <set>
  9 using namespace std;
 10 
 11 #define N 40005
 12 #define ll root<<1
 13 #define rr root<<1|1
 14 #define mid (a[root].l+a[root].r)/2
 15 
 16 
 17 int max(int x,int y){return x>y?x:y;}
 18 int min(int x,int y){return x<y?x:y;}
 19 int abs(int x,int y){return x<0?-x:x;}
 20 
 21 int n;
 22 int x[N];
 23 int m;
 24 
 25 int bin_s(int key){
 26     int l=1, r=m-1;
 27     while(l<=r){
 28         int mm=(l+r)/2;
 29         if(x[mm]==key) return mm;
 30         if(x[mm]>key) r=mm-1;
 31         else if(x[mm]<key) l=mm+1;
 32     }
 33 }
 34 
 35 struct Line{
 36     int l, r;
 37 }line[N];
 38 
 39 struct node{
 40     int l, r, val;
 41     bool f;
 42 }a[N*4];
 43 
 44 
 45 void build(int l,int r,int root){
 46     a[root].l=l;
 47     a[root].r=r;
 48     a[root].val=-1;
 49     if(l==r) return;
 50     build(l,mid,ll);
 51     build(mid+1,r,rr);
 52 }
 53 
 54 void down(int root){
 55     if(a[root].val>0&&a[root].l!=a[root].r){
 56         a[ll].val=a[rr].val=a[root].val;
 57         a[root].val=-1;
 58     }
 59 }
 60 
 61 void update(int l,int r,int val,int root){
 62     if(a[root].val==val) return;
 63     if(a[root].l==l&&a[root].r==r){
 64         a[root].val=val;
 65         return;
 66     }
 67     down(root);
 68     if(r<=a[ll].r) update(l,r,val,ll);
 69     else if(l>=a[rr].l) update(l,r,val,rr);
 70     else{
 71         update(l,mid,val,ll);
 72         update(mid+1,r,val,rr);
 73     }
 74     if(a[ll].val==a[rr].val&&a[ll].val>0) a[root].val=a[ll].val;
 75 }
 76 
 77 bool visited[N];
 78 int ans;
 79 
 80 void query(int root){
 81     if(a[root].val!=-1&&!visited[a[root].val]) {
 82         ans++;
 83         visited[a[root].val]=true;
 84         return;
 85     } 
 86     if(a[root].l==a[root].r)return ;
 87     down(root);
 88     query(ll);
 89     query(rr);
 90 }
 91 
 92 void out(int root){
 93     if(a[root].l==a[root].r) {
 94         printf("%d ",a[root].val);
 95         return;
 96     }
 97     down(root);
 98     out(ll);
 99     out(rr);
100 }
101 
102 main()
103 {
104     int t, i, j, k;
105     
106     cin>>t;
107     while(t--){
108         scanf("%d",&n);
109         k=0;
110         for(i=0;i<n;i++) {
111             scanf("%d %d",&line[i].l,&line[i].r);
112             x[++k]=line[i].l;
113             x[++k]=line[i].r;
114         }
115         sort(x+1,x+k);
116         k=unique(x+1,x+k+1)-x;   
117         m=k;
118         for(i=2;i<k;i++){
119             if(x[i]-x[i-1]>1) x[m++]=x[i]-1;
120         }
121         sort(x,x+m);
122         build(1,m,1);
123         for(i=0;i<n;i++){
124             int l=bin_s(line[i].l);
125             int r=bin_s(line[i].r);
126             update(l,r,i+1,1);
127             
128         }
129         memset(visited,false,sizeof(visited));
130         ans=0;
131         query(1);
132         printf("%d\n",ans);
133         //out(1);
134     }
135 }