POJ 2528（summer III X） 区间离散化，线段树区间更新

一直在用树状数组解题，线段树都有点忘了。。。。

参考了别人的代码后终于明白了离散化的道理和步骤，一下为参考网址：点击打开链接

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int casenum,postnum,ans,cnt,temp;
const int maxn=40000;
int vis[maxn];

int m[maxn][2];

struct Tree{
    int l;
    int r;
    int n;///n统计颜色
}tree[maxn];

struct Dis{
    int interval;///interval记录原来区间的边，num记录位置（方便区间对应）
    int num;
}dis[maxn];


bool cmp1(const Dis &a,const Dis &b)
{
    return a.interval<b.interval;
}

void init(int node,int l,int r)
{
    tree[node].l=l;
    tree[node].r=r;
    tree[node].n=0;
    if(l!=r)
    {
    int mid=(l+r)>>1;///除2是>>1,乘2是>>1;=(tree[node].切忌写成l+tree[node].r)>>1，是当前的中点
    init(2*node,l,mid);
    init(2*node+1,mid+1,r);
    }
}


void color(int col,int l,int r,int node)
{
    if(tree[node].l==l&&tree[node].r==r)
    {
        tree[node].n=col;///找到了区间，更新这个区间的颜色
        return ;
    }
    int mid=(tree[node].l+tree[node].r)>>1;
    if(tree[node].n>0))///重点注意，如果这个区间被访问了，并且这个区间有颜色，就要将这个区间的颜色更新到其左右孩子的节点，并且要将这个区间的颜色清空，这样才能算是覆盖
    {///区间被访问是指其区间的起点和终点被访问，所以中间的要清空
    {
        tree[node*2].n=tree[node*2+1].n=tree[node].n;
        tree[node].n=0;
    }
    if(tree[2*node+1].l<=l)///注意这里是区间子节点去匹配
        color(col,l,r,2*node+1);
    else if(tree[2*node].r>=r)
        color(col,l,r,2*node);///同上
    else
    {
        color(col,l,mid,2*node);
        color(col,mid+1,r,2*node+1);
    }
}

void solve(int node)
{
    if(tree[node].n)///如果这个区间有颜色了，停止访问并返回，因为下面的无论有没有颜色都是已经被覆盖的
    {
        if(!vis[tree[node].n])///如果有颜色且没被统计过的，就统计一次
        {
            ans++;
            vis[tree[node].n]=1;
        }
        return;
    }
    solve(2*node);
    solve(2*node+1);
    return;
}

int main()
{
    scanf("%d",&casenum);
    while(casenum--)
    {
        ans=0;
        cnt=0;
        scanf("%d",&postnum);
        for(int i=0;i<postnum;i++)
        {
            scanf("%d%d",&m[i][0],&m[i][1]);
            dis[2*i].interval=m[i][0];///这里直接数组相邻的两个位置记录区间值，而不用二维数组，方便后面遍历，也减少复杂度。
            dis[2*i+1].interval=m[i][1];
            dis[2*i].num=-(i+1);
            dis[2*i+1].num=i+1;
        }
        sort(dis,dis+2*postnum,cmp1);
        temp=dis[0].interval;
        cnt=1;///坐标从1开始
        
        
        ///离散化!!!!!!
        for(int i=0;i<2*postnum;i++)
        {
            if(temp!=dis[i].interval)
            {
                cnt++;///如果和前面的不同，这迭代加1
                temp=dis[i].interval;
            }
            if(dis[i].num>0)
            {
                m[dis[i].num-1][1]=cnt;///如map[0][0],map[0][1]分别代表排序后第一个张贴海报的左右区间，其中的数值为离散化后数轴的坐标
            }
            else
            {
                m[-dis[i].num-1][0]=cnt;///cnt为其离散化后所在数轴的坐标
            }
        }
        
        
        
        memset(tree,0,sizeof(tree));
        memset(vis,0,sizeof(vis));
        init(1,1,cnt);///一开始debug好久都没找到问题所在，原来位置参数的位置写错了
        for(int i=0;i<postnum;i++)
        {
            color(1+i,m[i][0],m[i][1],1);///每次贴的海报都不一样（这里想象着涂色，即每次涂得的颜色不一样）
        }
        solve(1);
        printf("%d\n",ans);
    }
}