您的位置:首页 > 理论基础 > 数据结构算法

zoj1610(线段树成段覆盖,lazy标记)

2013-12-03 15:37 393 查看
题目链接:zoj1610

/*zoj 1610 Count the Colors 线段树成段覆盖,lazy标记

题意:
给一段区间染色,后染的颜色会覆盖掉以前的颜色
问最后能看到的颜色以及有几段区间是这种颜色

思路:
需要注意的是对区间染色,而不是对点染色,
叶子结点的l和r表示的是区间的标号,点0~1为区间①,1~2为区间②
例:l = 1,r = 2,表示的是区间①和②,即点0~2之间的区间
*/
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define lson l, mid, rt<<1
#define rson mid+1, r, rt<<1|1
const int N = 8010;
int n,flag;
struct node
{
int l,r,col;
}s[N<<2];
int res
;//区间的颜色
int ans
;
void build(int l, int r, int rt)
{
s[rt].l = l;
s[rt].r = r;
s[rt].col = -1;
if(l == r) return ;
int mid = (l + r) >> 1;
build(lson);
build(rson);
}
void updata(int l, int r, int rt, int color)
{
if(s[rt].col == color) return;
if(l <= s[rt].l && s[rt].r <= r){
s[rt].col = color;
return;
}
if(s[rt].col != -1){//lazy标记
s[rt<<1].col = s[rt<<1|1].col = s[rt].col;
s[rt].col = -1;
}
int mid = (s[rt].l + s[rt].r) >> 1;
if(l <= mid) updata(l, r, rt<<1, color);
if(r > mid) updata(l, r, rt<<1|1, color);
}
void query(int rt)
{
if(s[rt].l == s[rt].r){
res[ s[rt].l ] = s[rt].col;
return;
}
if(s[rt].col != -1){
for(int i = s[rt].l; i <= s[rt].r; i ++)
res[i] = s[rt].col;
return;
}
query(rt<<1);
query(rt<<1|1);
}
int main()
{
int i,m,x,y,color;
while(~scanf("%d",&m))
{
build(1, N, 1);
for(i = 0; i < m; i ++){
scanf("%d%d%d",&x,&y,&color);
updata(x+1, y, 1, color);//注意,不是更新点,而是区间
}
memset(res, -1, sizeof(res));
memset(ans, 0, sizeof(ans));
query(1);
for(i = 1; i <= N;i ++){
if(res[i] != -1 && res[i] != res[i-1])
ans[ res[i] ] ++;
}
for(i = 0; i <= N; i ++)
if(ans[i])
printf("%d %d\n",i,ans[i]);
printf("\n");
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息