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);    }}