bzoj2243 [SDOI2011]染色

题目链接

树上区间修改，用树链剖分+线段树

只是这个题合并的时候不好有点坑，调了半天

然后不知道为什么用现成的树剖求lca居然要T，改了倍增才过

#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<string>
#include<cmath>
#include<ctime>
#include<queue>
#include<stack>
#include<map>
#include<set>
#define rre(i,r,l) for(int i=(r);i>=(l);i--)
#define re(i,l,r) for(int i=(l);i<=(r);i++)
#define Clear(a,b) memset(a,b,sizeof(a))
#define inout(x) printf("%d",(x))
#define douin(x) scanf("%lf",&x)
#define strin(x) scanf("%s",(x))
#define LLin(x) scanf("%lld",&x)
#define op operator
#define CSC main
typedef unsigned long long ULL;
typedef const int cint;
typedef long long LL;
using namespace std;
void inin(int &ret)
{
    ret=0;int f=0;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=1;ch=getchar();}
    while(ch>='0'&&ch<='9')ret*=10,ret+=ch-'0',ch=getchar();
    ret=f?-ret:ret;
}
int n,m,shen[100010],head[100010],next[200020],zhi[200020],col[100010],ed,son[100010],fa[100010],dfn[100010],tot,top[100010];
int ft[100010][18];
void dfs1(int x)
{
    son[x]=1;
    for(int i=1;i<=17;i++)
    {
        if(shen[x]<(1<<i))break;
        ft[x][i]=ft[ft[x][i-1]][i-1];
    }
    for(int i=head[x];i;i=next[i])if(zhi[i]!=ft[x][0])
    {
        shen[zhi[i]]=shen[x]+1;
        ft[zhi[i]][0]=x;
        dfs1(zhi[i]);
        son[x]+=son[zhi[i]];
    }
}
void dfs2(int x,int chain)
{
    dfn[x]=++tot;top[x]=chain;
    int k=0;
    for(int i=head[x];i;i=next[i])
        if(shen[zhi[i]]>shen[x]&&son[k]<son[zhi[i]])
           k=zhi[i];
    if(!k)return;
    dfs2(k,chain);
    for(int i=head[x];i;i=next[i])
        if(shen[zhi[i]]>shen[x]&&k!=zhi[i])
           dfs2(zhi[i],zhi[i]);
}
void add(int a,int b)
{
    next[++ed]=head[a],head[a]=ed,zhi[ed]=b;
    next[++ed]=head[b],head[b]=ed,zhi[ed]=a;
}
struct segtree
{
    int sum,lc,rc,l,r,add;
    segtree(){lc=rc=sum=0;}
}t[400040];
void build(int k,int l,int r)
{
    t[k].l=l;t[k].r=r;t[k].sum=1;t[k].add=0;
    if(l==r)return;
    int mid=(l+r)>>1;
    build(k<<1,l,mid);build(k<<1|1,mid+1,r);
}
void update(int k,int a,int x)
{
    int l=t[k].l,r=t[k].r,mid=(l+r)>>1,p1=k<<1,p2=p1|1;
    if(l==r)
    {
        t[k].sum=1,t[k].lc=t[k].rc=x;
        return ;
    }
    if(a<=mid)update(p1,a,x);
    else update(p2,a,x);
    t[k].sum=t[p1].sum+t[p2].sum;
    if(t[p1].rc==t[p2].lc)t[k].sum--;
    t[k].lc=t[p1].lc,t[k].rc=t[p2].rc;
}
void down(int k)
{
    if(!t[k].add)return ;
    int p1=k<<1,p2=p1|1;
    t[p1].add=t[p2].add=t[k].add;
    t[p1].sum=t[p2].sum=1;
    t[p1].lc=t[p1].rc=t[p2].lc=t[p2].rc=t[k].add;
    t[k].add=0;
    if(t[p1].l==t[p1].r)col[t[p1].l]=t[p1].add;
    if(t[p2].l==t[p2].r)col[t[p2].l]=t[p2].add;
}
void change(int k,int x,int y,int c)
{
    int l=t[k].l,r=t[k].r;
    if(l==x&&r==y)
    {t[k].lc=t[k].rc=c;t[k].sum=1;t[k].add=c;return;}
    down(k);
    int mid=(l+r)>>1;
    if(mid>=y)change(k<<1,x,y,c);
    else if(mid<x)change(k<<1|1,x,y,c);
    else
    {
        change(k<<1,x,mid,c);
        change(k<<1|1,mid+1,y,c);
    }
    t[k].lc=t[k<<1].lc;t[k].rc=t[k<<1|1].rc;
    if(t[k<<1].rc^t[k<<1|1].lc)t[k].sum=t[k<<1].sum+t[k<<1|1].sum;
    else t[k].sum=t[k<<1].sum+t[k<<1|1].sum-1;
}
int query(int k,int x,int y)
{
    int l=t[k].l,r=t[k].r;
    if(l==x&&r==y)return t[k].sum;
    down(k);
    int mid=(l+r)>>1;
    if(mid>=y)return query(k<<1,x,y);
    else if(mid<x)return query(k<<1|1,x,y);
    else
    {
        int tmp=1;
        if(t[k<<1].rc^t[k<<1|1].lc)tmp=0;
        return query(k<<1,x,mid)+query(k<<1|1,mid+1,y)-tmp;
    }
}
//int lca(int x,int y)
//{
//    while(top[x]!=top[y])
//    {
//        if(shen[top[x]]>shen[top[y]])x=fa[top[x]];
//        else y=fa[top[y]];
//    }
//    return shen[x]>shen[y]?y:x;
//}
int lca(int x,int y)
{
    if(shen[x]<shen[y])swap(x,y);
    int t=shen[x]-shen[y];
    for(int i=0;i<=17;i++)
       if(t&(1<<i))x=ft[x][i];
    for(int i=17;i>=0;i--)
       if(ft[x][i]!=ft[y][i])
       {x=ft[x][i];y=ft[y][i];}
    if(x==y)return x;
    return ft[x][0];
}
void change__(int x,int f,int c)
{
    while(top[x]!=top[f])
    {
        change(1,dfn[top[x]],dfn[x],c);
        x=ft[top[x]][0]; 
    }
    change(1,dfn[f],dfn[x],c);
}
int getc(int k,int x)
{
    int l=t[k].l,r=t[k].r;
    if(l==r)return t[k].lc;
    down(k);
    int mid=(l+r)>>1;
    if(x<=mid)return getc(k<<1,x);
    else return getc(k<<1|1,x);
}
int ad(int x,int f)
{
    int ret=0;
    while(top[x]!=top[f])
    {
        ret+=query(1,dfn[top[x]],dfn[x]);
        if(getc(1,dfn[top[x]])==getc(1,dfn[ft[top[x]][0]]))ret--;
        x=ft[top[x]][0]; 
    }
    ret+=query(1,dfn[f],dfn[x]);
    return ret;
}
int CSC()
{
    freopen("in.in","r",stdin);
    freopen("out.out","w",stdout);
    inin(n);inin(m);
    re(i,1,n)inin(col[i]);
    re(i,1,n-1)
    {
        int q,w;
        inin(q),inin(w);
        add(q,w);
    }
    dfs1(1);
    dfs2(1,1);
    build(1,1,n);
    for(int i=1;i<=n;i++)
        change(1,dfn[i],dfn[i],col[i]);
    re(i,1,m)
    {
        char opt[5];strin(opt);
        if(opt[0]=='C')
        {
            int q,w,e;
            inin(q),inin(w),inin(e);
            int r=lca(q,w);
            change__(q,r,e),change__(w,r,e);
        }
        if(opt[0]=='Q')
        {
            int q,w;
            inin(q),inin(w);
            int e=lca(q,w);
            printf("%d
",ad(q,e)+ad(w,e)-1);
        }
    }
    return 0;
}