zoj 1610 Count the Colors(线段树 区间更新)

题意：

给c1~c2的区间涂色。询问最后有几段颜色区域。初始无色。

思路：

lazy标记，区间逐渐更新。

自己最初写了一个从后往前的仅仅带标记的线段树，竟然WA了。很迷。只好规规矩矩来一遍，终于过了

#include <stdio.h>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn=8050;
struct node
{
int left,right,val,lazy ;
}tree[4*maxn];

int n;
int color[maxn*4];
int ans[maxn*4];
int l[maxn],r[maxn],c[maxn];
int cnt=0;
void Build(int i,int left,int right)
{
tree[i].lazy=0;
tree[i].val=-1;
tree[i].left=left;
tree[i].right=right;
if(left==right)
{
return ;
}
int mid=(tree[i].left+tree[i].right)>>1;
Build(i<<1,left,mid);
Build(i<<1|1,mid+1,right);
return ;
}
void update(int i,int left,int right,int w)
{
if(tree[i].left==left&&tree[i].right==right)
{
tree[i].lazy=1;
tree[i].val=w;
return ;
}
if(tree[i].lazy)
{
tree[i].lazy=0;
tree[i<<1].lazy=1,tree[i<<1|1].lazy=1;
tree[i<<1].val=tree[i].val,tree[i<<1|1].val=tree[i].val;
}
int mid=(tree[i].left+tree[i].right)>>1;
if(left>mid)
{
update(i<<1|1,left,right,w);
}
else if(right<=mid)
{
update(i<<1,left,right ,w);
}
else
{
update(i<<1,left,mid,w);
update(i<<1|1,mid+1,right,w);
}
return ;
}
void query(int i)
{
if(tree[i].lazy)
{
for(int j=tree[i].left;j<=tree[i].right;j++)
color[j]=tree[i].val;
return ;
}
if(tree[i].left==tree[i].right)
return ;
int mid=(tree[i].left+tree[i].right)>>1;
query(i<<1);
query(i<<1|1);
}
int main()
{
while(~scanf("%d",&n))
{
cnt=0;
memset(color,-1,sizeof(color));
memset(tree,0,sizeof(tree));
memset(ans,0,sizeof(ans));
Build(1,1,maxn);
int num=0;
for(int i=1;i<=n;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
update(1,u+1,v,w);
}
query(1);
for(int i=1;i<=maxn;i++)
{
if(color[i]!=color[i-1]&&color[i]!=-1)
{
ans[color[i]]++;
}
}
for(int i=0;i< maxn;i++)
if(ans[i]!=0)
printf("%d %d\n",i,ans[i]);
printf("\n");
}
}
/*
3
1 2 0
0 1 0
2 3 0
*/

下面的代码Wa了。求帮助

#include <stdio.h>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn=8050;
struct node
{
int left,right,val,col;
}tree[4*maxn];
int n;
int color[maxn*4];
int ans[maxn*4];
int l[maxn],r[maxn],c[maxn];
int cnt=0;
void Build(int i,int left,int right)
{
tree[i].left=left;
tree[i].right=right;
if(left==right)
{
tree[i].col=-1;
tree[i].val=0;
return ;
}
int mid=(tree[i].left+tree[i].right)>>1;
Build(i<
4000
<1,left,mid);
Build(i<<1|1,mid+1,right);
tree[i].val=tree[i<<1].val&&tree[i<<1|1].val;
return ;
}

void update(int i,int left,int right,int w)
{
if(tree[i].val==1 )
{
return ;
}
if(tree[i].left==left&&tree[i].right==right)
{
tree[i].val=1;
tree[i].col=w;
for(int j=tree[i].left;j<=tree[i].right;j++)
color[j]=w;
return ;
}
int mid=(tree[i].left+tree[i].right)>>1;
if(left>mid)
{
update(i<<1|1,left,right,w);
}
else if(right<=mid)
{
update(i<<1,left,right ,w);
}
else
{
update(i<<1,left,mid,w);
update(i<<1|1,mid+1,right,w);
}
tree[i].val=tree[i<<1].val&&tree[i<<1|1].val;
return ;
}

int main()
{
while(~scanf("%d",&n))
{
cnt=0;
memset(color,-1,sizeof(color));
memset(tree,0,sizeof(tree));
memset(ans,0,sizeof(ans));
Build(1,1,maxn);
int num=0;
for(int i=1;i<=n;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
if(u>=v)
continue;
l[++num]=++u,r[num]=v,c[num]=w;
}
for(int i=num;i>=1;i--)
{
update(1,l[i],r[i],c[i]);
}
for(int i=1;i<=maxn;i++)
{
if(color[i]!=color[i-1]&&color[i]!=-1)
{
ans[color[i]]++;
}
}
for(int i=0;i<=maxn;i++)
if(ans[i]!=0)
printf("%d %d\n",i,ans[i]);
printf("\n");
}
}
/*
3
1 2 0
0 1 0
2 3 0
*/