当前位置: 代码迷 >> 综合 >> POJ 2528(summer III X) 区间离散化,线段树区间更新
  详细解决方案

POJ 2528(summer III X) 区间离散化,线段树区间更新

热度:95   发布时间:2023-12-16 03:54:43.0

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

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

#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++;///如果和前面的不同,这迭代加1temp=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);}
}


  相关解决方案