POJ 2528 Mayor's posters（线段树染色问题+离散化）

http://poj.org/problem?id=2528

题意：

给出一面无限长的墙，现在往墙上依次贴海报，问最后还能看见多少张海报。

题意：
这道题目就相当于对x轴染色，然后计算出最后还能看见多少种颜色。

由于数据量是给定的，所以需要进行离散化。但是这道题目的话，不能简单的离散化，前后相差大于1的话需要额外的插一个数。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<sstream>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const int maxn = 10000 + 5;

int n;
int ans;
int col[maxn<<4];
int vis[maxn<<4];
int l_pos[maxn<<2],r_pos[maxn<<2];
int x[maxn<<4];

void PushDown(int o)
{
    if(col[o]!=-1)
    {
        col[o<<1]=col[o<<1|1]=col[o];
        col[o]=-1;
    }
}

void update(int L, int R, int l, int r, int ID, int o)
{
    if(L<=l && R>=r)
    {
        col[o]=ID;
        return;
    }
    PushDown(o);
    int mid=(l+r)/2;
    if(L<=mid)  update(L,R,l,mid,ID,o<<1);
    if(R>mid)   update(L,R,mid+1,r,ID,o<<1|1);
}

void query(int L, int R, int l, int r, int o)
{
    if(col[o]!=-1)
    {
        if(vis[col[o]]==0)  //每种颜色统计一次
        {
            ans++;
            vis[col[o]]=1;
        }
        col[o]=-1;
        return;
    }
    if(l==r)  return;
    PushDown(o);
    int mid=(l+r)/2;
    if(L<=mid)  query(L,R,l,mid,o<<1);
    if(R>mid)   query(L,R,mid+1,r,o<<1|1);
}

int main()
{
    //freopen("in.txt","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(col,-1,sizeof(col));
        memset(vis,0,sizeof(vis));

        scanf("%d",&n);
        int cnt=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&l_pos[i],&r_pos[i]);
            x[++cnt]=l_pos[i];
            x[++cnt]=r_pos[i];
        }

        sort(x+1,x+cnt+1);
        int cntt=cnt;
        for(int i=2;i<=cntt;i++)
        {
            if(x[i]-x[i-1]>1)   x[++cnt]=x[i-1]+1;  //插点
        }
        //离散化
        sort(x+1,x+cnt+1);
        int num=unique(x+1,x+cnt+1)-(x+1);
        for(int i=1;i<=n;i++)
        {
            l_pos[i]=lower_bound(x+1,x+num+1,l_pos[i])-(x+1);
            r_pos[i]=lower_bound(x+1,x+num+1,r_pos[i])-(x+1);
            update(l_pos[i],r_pos[i],0,num,i,1);
        }

        ans=0;
        query(0,num,0,num,1);
        printf("%d
",ans);
    }
    return 0;
}