[SDOI2011]染色 bzoj 2243 树链剖分

题目大意

给定一棵有n个节点的无根树和m个操作，操作有2类：
1、将节点a到节点b路径上所有点都染成颜色c；
2、询问节点a到节点b路径上的颜色段数量（连续相同颜色被认为是同一段），
如“112221”由3段组成：“11”、“222”和“1”。
请你写一个程序依次完成这m个操作。

分析

打了大概2个半小时的代码，真真正正的码字题。
复习了树链剖分的写法和学习了一下新的知识。
这题主要是线段树建的时候要建好

code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>
#define maxn 200005  
#define maxm 300005 
#define INF 0x7fffffff;
using namespace std;

struct gg{
    int x,y;
    int next;
}edge[maxm];
int ls[maxn];
int edge_m;

struct arr{
    int x,y;
    int l,r;
    int lazy,num;
}f[maxm*6];

int a[maxn];
int n,m;

void swap1(int &x,int &y)
{
    int z;
    z=x; x=y; y=z;
    return;
}

void add(int x,int y)
{
    edge[++edge_m]=(gg){x,y,ls[x]},ls[x]=edge_m;
    edge[++edge_m]=(gg){y,x,ls[y]},ls[y]=edge_m;
}

void delay(int x)//延迟标记。
{
    if (f[x].lazy==0) return;
    f[x*2].lazy=f[x].lazy; f[x*2+1].lazy=f[x].lazy;
    f[x*2].l=f[x].lazy; f[x*2].r=f[x].lazy;
    f[x*2+1].l=f[x].lazy; f[x*2+1].r=f[x].lazy;
    if (f[x].lazy!=0) f[x*2].num=1,f[x*2+1].num=1;
    f[x].lazy=0;
}

int insert(int r,int x,int y,int add)//线段树修改操作。
{
    delay(r);
    if ((f[r].x==x)&&(f[r].y==y)) 
    {
        f[r].num=1;
        f[r].l=add; f[r].r=add;
        f[r].lazy=add;
        return 0;
    }
    int mid=(f[r].x+f[r].y)/2;
    if (y<=mid) insert(r*2,x,y,add);
        else if (x>mid) insert(r*2+1,x,y,add);
            else insert(r*2,x,mid,add),insert(r*2+1,mid+1,y,add);
    f[r].num=f[r*2].num+f[r*2+1].num;
    if (f[r*2].r==f[r*2+1].l) f[r].num--;
    f[r].l=f[r*2].l; f[r].r=f[r*2+1].r;
}

int maketree(int r,int x,int y)//线段树建树
{
    f[r].x=x; f[r].y=y; 
    f[r].num=0; f[r].l=0; f[r].r=0;
    if (x==y) return 0;
    int mid=(x+y)/2;
    maketree(r*2,x,mid);
    maketree(r*2+1,mid+1,y);
}

int siz[maxn],dep[maxn],top[maxn],fa[maxn],son[maxn],w[maxn];

void dfs1(int x,int r)//树链dfs1 
{
    fa[x]=r;
    siz[x]=1;
    dep[x]=dep[r]+1;
    int mx=0;
    for (int i=ls[x];i;i=edge[i].next)
    {
        if (edge[i].y==r) continue;
        dfs1(edge[i].y,x);
        siz[x]+=siz[edge[i].y];
        if (mx<siz[edge[i].y])
        {
            mx=siz[edge[i].y];
            son[x]=edge[i].y;
        }
    }
    return;
}

int num=0;

void dfs2(int x,int st)//树链dfs2 
{
    num++;
    w[x]=num; top[x]=st;
    if (son[x]!=0)  dfs2(son[x],st);
    for (int i=ls[x];i;i=edge[i].next)
        if ((edge[i].y!=fa[x])&&(edge[i].y!=son[x]))
            dfs2(edge[i].y,edge[i].y);
    return;
}

int findsum(int r,int x,int y)//线段树 
{
    delay(r);
    if ((f[r].x==x)&&(f[r].y==y)) return f[r].num;
    int mid=(f[r].x+f[r].y)/2;
    if (y<=mid) return findsum(r*2,x,y);
        else if (x>mid) return findsum(r*2+1,x,y);
    int k=findsum(r*2,x,mid)+findsum(r*2+1,mid+1,y);
    if (f[r*2].r==f[r*2+1].l) k--;
    return k;
}

int findcolor(int r,int x)//线段树 
{
    delay(r);
    if ((f[r].x==x)&&(f[r].y==x)) return f[r].l;
    int mid=(f[r].x+f[r].y)/2;
    if (x<=mid) return findcolor(r*2,x);
        else return findcolor(r*2+1,x);
} 

int solvesum(int x,int y)//树链 
{
    int mx=0;
    int f1=top[x];
    int f2=top[y];
    while (f1!=f2)
    {
        if (dep[f1]<dep[f2]) 
        {swap1(x,y); swap1(f1,f2);};
        mx+=findsum(1,w[f1],w[x]);
        if (findcolor(1,w[f1])==findcolor(1,w[fa[f1]])) mx--;
        x=fa[f1]; f1=top[x];
    }
    if (dep[x]>dep[y]) swap1(x,y);
    mx+=findsum(1,w[x],w[y]);
    return mx;
}

void solvein(int x,int y,int z)//树链 
{
    int f1=top[x];
    int f2=top[y];
    while (f1!=f2)
    {
        if (dep[f1]<dep[f2]) 
        {swap1(x,y); swap1(f1,f2);};
        insert(1,w[f1],w[x],z);
        x=fa[f1]; f1=top[x];
    }
    if (dep[x]>dep[y]) swap1(x,y);
    insert(1,w[x],w[y],z);
}

void init()
{
    scanf("%d%d\n",&n,&m);
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for (int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
    }
}

int main()
{
    init();
    maketree(1,1,maxn*2);
    dfs1(1,0);
    dfs2(1,1);
    for (int i=1;i<=n;i++)
        insert(1,w[i],w[i],a[i]);
    char c[10];
    int x,y,z;
    for (int i=1;i<=m;i++)
    {
        scanf("\n");
        scanf("%s%d%d",c,&x,&y);
        if (c[0]=='C') 
        {
            scanf("%d",&z);
            solvein(x,y,z);
        }
        else
            if (c[0]=='Q')
                printf("%d\n",solvesum(x,y));
    }
    return 0;
}