大二时曾接触过线段树,知道大概是什么样子,会写最简单的那种,比如区间最值,区间和,但也是局限在一个点更新的情况下,现在接触到了lazy思想,我认为lazy思想才是线段树的精华吧,而且不是那么好理解。
首先说下,复杂度的问题,因为一条任意的线段最多被分解为2logn个线段,所以我们查询的复杂度是logn,更新一个点的时候,我们其实也就是做树高的复杂度就好了,所以也是logn,但是如果是多个点一起更新呢?这里lazy思想就出现了,我们更新的时候只更新必要的,然后在该节点说明一下,说明这一点以下都是没更新过,如果你要往下查询或者往下更新,都需要把没更新的信息传到该节点的左右儿子,接着递归它的左右儿子。值得注意的一点是,这个lazy表示是非常巧妙的,不同的问题需要设置的lazy标记都不太一样,这也是lazy思想最难的地方。下面说明poj一道线段树基础题(需要lazy)。
poj2777
这题需要一个位运算通过左右儿子的颜色值来计算父亲节点的颜色值,有lazy标记之后,更新和查找的时候都会遇到lazy点和非lazy点,更新时,有2种情况,1我遇到了刚好的更新区间,那么我不用往下更新了,我在当前点做一个lazy=1,然后把当前点的信息更新下。2我发现区间不是刚刚好的,这是我需要分解开来更新,此时又有2种情况,(2.1)这个点的lazy是1,那么说明我下面的点都是没更新过的,那么我先把他的左右儿子先更新了(需要更新2个lazy,2个color,把自己lazy置0,5条语句),然后看看我要怎么分解,分解也是按照都在左边,都在右边,左右都有3种情况,假如只有一边,另外一边怎么处理?答案是需要调用getc()函数得到另外一边的颜色值,两者做一个或运算,作为当前节点更新后的值。(2.2)这个点的lazy是0,那么好,最起码当前点的左右儿子是被更新过的,我不用先更新这个点的左右儿子了,直接去递归左右儿子就好。
其实一句话,如何递归左右儿子都是一样的,只是发现lazy=1点时,一定要先更新他的儿子再去递归他的左右儿子就好了,lazy=0,直接递归。
写计算颜色种数函数也差不多,就是说,lazy=1的时候也需要更新儿子节点。再递归。这里不多说了。
poj2777代码如下:
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
struct node
{
int l,r,lazy,co;
}tr[500001];
void cre(int l,int r,int node)
{
tr[node].l=l;
tr[node].r=r;
tr[node].lazy=0;
tr[node].co=0;
if(l==r)
return ;
int mid=(l+r)/2;
cre(l,mid,node*2);
cre(mid+1,r,node*2+1);
return ;
}
int getc(int l,int r,int node)
{
if(l==tr[node].l&&tr[node].r==r)
return tr[node].co;
if(tr[node].lazy)
{
tr[2*node].co=tr[node].co;
tr[2*node+1].co=tr[node].co;
tr[2*node].lazy=1;
tr[2*node+1].lazy=1;
tr[node].lazy=0;
}
int mid=(tr[node].l+tr[node].r)/2;
int c1,c2,c;
if(mid>=r)
c=getc(l,r,2*node);
else if(mid<l)
c=getc(l,r,2*node+1);
else
{
c1=getc(l,mid,2*node);
c2=getc(mid+1,r,2*node+1);
c=c1|c2;
}
return c;
}
int upd(int l,int r,int col,int node)
{
if(l==tr[node].l&&tr[node].r==r)
{
tr[node].lazy=1;
tr[node].co=col;
return tr[node].co;
}
if(tr[node].lazy)
{
tr[2*node].co=tr[node].co;
tr[2*node+1].co=tr[node].co;
tr[2*node].lazy=1;
tr[2*node+1].lazy=1;
tr[node].lazy=0;
}
int mid=(tr[node].l+tr[node].r)/2;
int c1,c2;
if(mid>=r)
{
c1=upd(l,r,col,2*node);
c2=getc(mid+1,tr[node].r,2*node+1);
}
else if(mid<l)
{
c1=getc(tr[node].l,mid,2*node);
c2=upd(l,r,col,2*node+1);
}
else
{
c1=upd(l,mid,col,2*node);
c2=upd(mid+1,r,col,2*node+1);
}
tr[node].co=c1|c2;
return tr[node].co;
}
int n,c,o;
int main()
{
while(~scanf("%d%d%d",&n,&c,&o))
{
cre(1,n,1);
for(int i=1;i<=n;i++)
upd(i,i,1,1);
while(o--)
{
char cc[2];
int a,b,color;
scanf("%s%d%d",cc,&a,&b);
if(a>b)
swap(a,b);
if(cc[0]=='C')
{
scanf("%d",&color);
upd(a,b,1<<(color-1),1);
}
else if(cc[0]=='P')
{
int x=getc(a,b,1);
int ans=0;
while(x)
{
x&=(x-1);
ans++;
}
printf("%d\n",ans);
}
}
}
return 0;
}
<span style="font-size:18px"></span>