树链剖分+线段树【SDOI2011】 bzoj2243 染色

题目大意：

给一棵树，每个节点有一个颜色。写一个程序支持把两个点路径上的所有点染成一个颜色，查询两点之间的色段数量。

解题思路：

树链剖分+线段树

首先它是一颗树，而且是修改和查询两点路径上的颜色，可以想到树链剖分。

查询颜色段数可以用线段树维护区间颜色段数。

这道题涉及到区间合并，所以在线段树和lca的时候需要多记录一些东西，当前区间的最左边的颜色，最右边的颜色，已经求出的区间的左边的颜色。

我采用的做法是先求出lca，然后从左右两边各爬一边，就能有序的将所有的区间连在一起，当然大神告诉我可以左右同时爬，但是我不太会。

注意事项：

1、在树链剖分求lca的时候一定要写函数接口，不要把大段代码放在lca里，代码非常凌乱，不易差错；

2、打了标记要记得下传，时时刻刻思考要不要下传标记；

3、每次区间合并的时候要判断是否连接的部分是相同的颜色，如果相同要ans--；

4、从一侧爬的时候处理lca，另一侧就要避开lca，而且左右两条链合并的时候要判断颜色是否相同。

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<string>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 100100
using namespace std;
struct tree{
    int l,r,sum,ls,rs,lz;
}d[N*8];
int n,m,x,y,z,a,b;
char s[5];
int q[N];
int fir[N],nes[N*2],v[N*2];
bool pc[N*2];
int dep[N],fa[N],zon[N],sz[N],ld[N],xl[N],pos[N],top;
void dfs1(int c)
{
    dep[c]=dep[fa[c]]+1;
    sz[c]=1;
    for(int t=fir[c];t;t=nes[t])
    {
        if(pc[t]) continue;
        pc[t^1]=true;
        fa[v[t]]=c;
        dfs1(v[t]);
        sz[c]+=sz[v[t]];
        if(sz[v[t]]>sz[zon[c]]) zon[c]=v[t];
    }
}
void dfs2(int c)
{
    xl[++top]=c;
    pos[c]=top;
    ld[c]=c;
    if(zon[fa[c]]==c) ld[c]=ld[fa[c]];
    if(zon[c]) dfs2(zon[c]);
    for(int t=fir[c];t;t=nes[t])
    {
        if(pc[t] || v[t]==zon[c]) continue;
        dfs2(v[t]);
    }
}
void make_tree(int c,int l,int r)
{
    d[c].l=l;d[c].r=r;
    if(l==r)
    {
        d[c].sum=1;
        d[c].ls=d[c].rs=q[xl[l]];
        d[c].lz=-1;
        return;
    }
    int mid=(l+r)/2;
    make_tree(c*2,l,mid);
    make_tree(c*2+1,mid+1,r);
    d[c].sum=d[c*2].sum+d[c*2+1].sum;
    if(d[c*2].rs==d[c*2+1].ls) d[c].sum--;
    d[c].ls=d[c*2].ls;
    d[c].rs=d[c*2+1].rs;
    d[c].lz=-1;
}
void push_down(int c)
{
    if(d[c].lz==-1) return;
    d[c].sum=1;
    d[c].ls=d[c].rs=d[c].lz;
    d[c*2].lz=d[c*2+1].lz=d[c].lz;
    d[c].lz=-1;
    return;
}
void change(int c)
{
    if(a<=d[c].l && b>=d[c].r) {d[c].lz=z;return;}
    push_down(c);
    int mid=(d[c].l+d[c].r)/2;
    if(a<=mid) change(c*2);
    if(b>mid)  change(c*2+1);
    push_down(c*2);
    push_down(c*2+1);
    d[c].sum=d[c*2].sum+d[c*2+1].sum;
    if(d[c*2].rs==d[c*2+1].ls) d[c].sum--;
    d[c].ls=d[c*2].ls;
    d[c].rs=d[c*2+1].rs;
    d[c].lz=-1;
}
void lca_change(int x,int y)
{
    while(ld[x]!=ld[y])
    {
        if(dep[ld[x]]<dep[ld[y]]) swap(x,y);
        a=pos[ld[x]]; b=pos[x];
        change(1);
        x=fa[ld[x]];
    }
    if(dep[x]<dep[y]) swap(x,y);
    a=pos[y]; b=pos[x];
    change(1);
}
int query(int c,int &zs,int &ys)
{
    push_down(c);
    if(a<=d[c].l && b>=d[c].r)
    {
        zs=d[c].ls; ys=d[c].rs;
        return d[c].sum;
    }
    int mid=(d[c].l+d[c].r)/2;
    if(b<=mid) return query(c*2,zs,ys);
    if(a>mid)  return query(c*2+1,zs,ys);
    int zz,yy,ans=0;
    ans+=query(c*2,zs,yy);
    ans+=query(c*2+1,zz,ys);
    if(d[c*2].rs==d[c*2+1].ls) ans--;
    return ans;
}
int lca_query(int x,int y)
{
    int lx=x,ly=y;
    while(ld[lx]!=ld[ly])
    {
        if(dep[ld[lx]]<dep[ld[ly]]) swap(lx,ly);
        lx=fa[ld[lx]];
    }
    if(dep[lx]<dep[ly]) swap (lx,ly);
    int lca=ly,ans=0;
    int ls=-1,zs=-1,ys=-1,k1,k2;
    while(ld[x]!=ld[lca])
    {
        a=pos[ld[x]];b=pos[x];
        ans+=query(1,zs,ys);
        if(ls==ys) ans--;
        ls=zs;
        x=fa[ld[x]];
    }
    a=pos[lca];b=pos[x];
    ans+=query(1,zs,ys);
    if(ls==ys) ans--;
    k1=zs;
    ls=zs=ys=-1;
    while(ld[y]!=ld[lca])
    {
        a=pos[ld[y]];b=pos[y];
        ans+=query(1,zs,ys);
        if(ls==ys) ans--;
        ls=zs;
        y=fa[ld[y]];
    }
    if(y!=lca)
    {
        a=pos[lca]+1; b=pos[y];
        ans+=query(1,zs,ys);
        if(ls==ys) ans--;
    }
    k2=zs;
    if(k1==k2) ans--;
    return ans;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&q[i]);
    for(int i=2;i<2*n;i+=2)
    {
        scanf("%d%d",&x,&y);
        nes[i]=fir[x];
        fir[x]=i; v[i]=y;
        nes[i^1]=fir[y];
        fir[y]=i^1; v[i^1]=x;
    }
    dfs1(1);dfs2(1);
    make_tree(1,1,n);
    for(int i=1;i<=m;i++)
    {
        scanf("%s",s);
        if(s[0]=='Q')
        {
            scanf("%d%d",&x,&y);
            printf("%d\n",lca_query(x,y));
        }
        else
        {
            scanf("%d%d%d",&x,&y,&z);
            lca_change(x,y);
        }
    }
    return 0;
}