标签:des style blog color io os ar for sp
Description
西南某科技大学的校门外长度为 L 的公路上有一排树,每两棵相邻的树之间的间隔都是 1 米。我们可以把马路看成一个数轴,马路的一端在数轴 1 的位置,另一端在 L 的位置;数轴上的每个整数点,即 1,2,……,L,都种有一棵树。
现在要将这排树的某一段涂成某种颜色,给定 N 组区间[ J,K ],表示从 J 到 K 的树都要涂上一种颜色(每次涂的颜色不一样,而且每次涂色会覆盖掉原来的颜色),区间之间可能有重叠的部分, L 的长度为 1000,0000。现在你的任务是计算涂色工作结束后,这一排树里总共有多少种颜色的树?(没涂过色的不算)
Input
第一行为一个整数N (1<=N<=10000)。 接下来的N行每行是两个整数数J,K(1<=J<=K<=1000,0000),表示涂色的起止区间。
多组测试数据。
Output
一个整数,表示有多少种颜色的树。Sample Input
2
1 10
5 5
1
1 1
Sample Output
2
1
Source
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> using namespace std; #define N 20005 struct tree { int l,r; }p[N<<2]; struct node { int val,id,pos; }h[N<<2]; int ans; int color[N<<2]; bool flag[N<<2]; int change[N<<2]; bool cmp(node a,node b) { return a.val<b.val; } void pushup(int rt) { color[rt]=0; } void pushdown(int rt) { if(change[rt]!=-1) { change[rt<<1]=change[rt<<1|1]=change[rt]; color[rt<<1]=color[rt<<1|1]=change[rt]; change[rt]=-1; } } void build(int l,int r,int rt) { color[rt]=0; change[rt]=-1;//初始化 if(l==r) return; int m=(l+r)>>1; build(l,m,rt<<1); build(m+1,r,rt<<1|1); } void update(int l,int r,int rt,int L,int R,int c) { if(l==L && R==r) { color[rt]=c; change[rt]=c; return; } pushdown(rt); int m=(l+r)>>1; if(R<=m) update(l,m,rt<<1,L,R,c); else if(L>m) update(m+1,r,rt<<1|1,L,R,c); else { update(l,m,rt<<1,L,m,c); update(m+1,r,rt<<1|1,m+1,R,c); } pushup(rt); } void query(int l,int r,int rt) { int m=(l+r)>>1; if(color[rt]) { if(!flag[color[rt]]) { ans++; flag[color[rt]]=1; } return; } if(l==r) return; //忘了加这一句、一直RE query(l,m,rt<<1); query(m+1,r,rt<<1|1); } int main() { int n,i; while(scanf("%d",&n)!=EOF) { ans=0; memset(flag,0,sizeof(flag)); for(i=0;i<n;i++) { scanf("%d%d",&p[i].l,&p[i].r); h[i*2].val=p[i].l; h[i*2].id=i; h[i*2].pos=0; h[i*2+1].val=p[i].r; h[i*2+1].id=i; h[i*2+1].pos=1; } /* 离散化 */ sort(h,h+n*2,cmp); int last=-1,total=0; for(i=0;i<n*2;i++) { if(h[i].val!=last) { total++; if(i && h[i].val-h[i-1].val>1) total++; //注意这里、= = last=h[i].val; } if(h[i].val!=total) { if(h[i].pos) p[h[i].id].r=total; else p[h[i].id].l=total; } } build(1,total,1); for(i=0;i<n;i++) { update(1,total,1,p[i].l,p[i].r,i+1); } query(1,total,1); cout<<ans<<endl; } return 0; }
线段树 [SWUST OJ 764] 校门外的树 Plus Plus
标签:des style blog color io os ar for sp
原文地址:http://www.cnblogs.com/hate13/p/4047820.html