BZOJ 3531 SDOI2014 旅行 树链剖分+线段树动态开点

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=3531

题意概述：
　　给出一棵N个点的树，树上的每个结点有一个颜色和权值，支持以下四种操作：
　　1.将点x的颜色改成c。
　　2.将点x的权值给成w。
　　3.询问x->y路径上和端点相同颜色的点的权值总和（端点颜色相同）。
　　4.询问x->y路径上和端点相同颜色的点的权值最大值（端点颜色相同）。

分析：

　　首先树链剖分，每条链开C棵维护每个颜色的线段树。动态开点。

　　感觉没什么好说的......考试的时候灵光一闪就来了......之前还在YYlct之类的......

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<cctype>
using namespace std;
const int maxn=100005;
const int SIZE=4000005;

int N,Q,C[maxn],W[maxn];
map<int,int>rt[maxn];
struct edge{ int to,next; }E[maxn<<1];
int first[maxn],np,dep[maxn],top[maxn],fa[maxn],son[maxn],sz[maxn],len[maxn];
int lc[SIZE],rc[SIZE],sum[SIZE],mx[SIZE],np2;

void _scanf(int &x)
{
    x=0;
    char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
}
void _scanf(char *s)
{
    int cnt=0;
    char ch=getchar();
    while(!isalpha(ch)) ch=getchar();
    while(isalpha(ch)) s[cnt++]=ch,ch=getchar();
    s[cnt]='';
}
int out_cnt,out[15];
void _printf(int x)
{
    out[++out_cnt]=x%10,x/=10;
    while(x) out[++out_cnt]=x%10,x/=10;
    while(out_cnt) putchar('0'+out[out_cnt--]);
    putchar('
');
}
void add_edge(int u,int v){
    E[++np]=(edge){v,first[u]};
    first[u]=np;
}
void data_in()
{
    _scanf(N);_scanf(Q);
    for(int i=1;i<=N;i++) _scanf(W[i]),_scanf(C[i]);
    int x,y;
    for(int i=1;i<N;i++){
        _scanf(x);_scanf(y);
        add_edge(x,y); add_edge(y,x);
    }
}
void DFS1(int i,int f,int d){
    fa[i]=f,dep[i]=d,sz[i]=1;
    for(int p=first[i];p;p=E[p].next){
        int j=E[p].to;
        if(j==f) continue;
        DFS1(j,i,d+1);
        sz[i]+=sz[j];
        if(sz[j]>sz[son[i]]) son[i]=j;
    }
}
void DFS2(int i,int f,int tp){
    top[i]=tp,len[tp]++;
    if(son[i]) DFS2(son[i],i,tp);
    for(int p=first[i];p;p=E[p].next){
        int j=E[p].to;
        if(j==f||j==son[i]) continue;
        DFS2(j,i,j);
    }
}
int newnode(){
    np2++,lc[np2]=rc[np2]=0,sum[np2]=mx[np2]=0;
    return np2;
}
void pushup(int now){
    mx[now]=max(mx[lc[now]],mx[rc[now]]);
    sum[now]=sum[lc[now]]+sum[rc[now]];
}
void update(int &now,int L,int R,int pos,int v){
    if(!now) now=newnode();
    if(L==R){ sum[now]=mx[now]=v; return; }
    int m=L+R>>1;
    if(pos<=m) update(lc[now],L,m,pos,v);
    else update(rc[now],m+1,R,pos,v);
    pushup(now);
}
void build(int i,int f){
    update(rt[top[i]][C[i]],0,len[top[i]]-1,dep[i]-dep[top[i]],W[i]);
    for(int p=first[i];p;p=E[p].next){
        if(E[p].to==f) continue;
        build(E[p].to,i);
    }
}
int query1(int now,int L,int R,int A,int B){
    if(!now) return 0;
    if(A<=L&&R<=B) return sum[now];
    int m=L+R>>1;
    if(B<=m) return query1(lc[now],L,m,A,B);
    if(A>m) return query1(rc[now],m+1,R,A,B);
    return query1(lc[now],L,m,A,B)+query1(rc[now],m+1,R,A,B);
}
int query2(int now,int L,int R,int A,int B){
    if(!now) return 0;
    if(A<=L&&R<=B) return mx[now];
    int m=L+R>>1;
    if(B<=m) return query2(lc[now],L,m,A,B);
    if(A>m) return query2(rc[now],m+1,R,A,B);
    return max(query2(lc[now],L,m,A,B),query2(rc[now],m+1,R,A,B));
}
int LCA(int x,int y){
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        x=fa[top[x]];
    }
    return dep[x]<dep[y]?x:y;
}
int query_s(int x,int y,int c){
    int re=0;
    while(top[x]!=top[y]){
        re+=query1(rt[top[x]][c],0,len[top[x]]-1,0,dep[x]-dep[top[x]]);
        x=fa[top[x]];
    }
    re+=query1(rt[top[x]][c],0,len[top[x]]-1,dep[y]-dep[top[x]],dep[x]-dep[top[x]]);
    return re;
}
int query_m(int x,int y,int c){
    int re=0;
    while(top[x]!=top[y]){
        re=max(re,query2(rt[top[x]][c],0,len[top[x]]-1,0,dep[x]-dep[top[x]]));
        x=fa[top[x]];
    }
    re=max(re,query2(rt[top[x]][c],0,len[top[x]]-1,dep[y]-dep[top[x]],dep[x]-dep[top[x]]));
    return re;
}
void work()
{
    DFS1(1,0,1);
    DFS2(1,0,1);
    build(1,0);
    char op[5];
    int x,y,z,p,c,w,ans;
    for(int i=1;i<=Q;i++){
        _scanf(op);
        if(op[0]=='C'&&op[1]=='C'){
            _scanf(x);_scanf(c);
            update(rt[top[x]][C[x]],0,len[top[x]]-1,dep[x]-dep[top[x]],0);
            update(rt[top[x]][C[x]=c],0,len[top[x]]-1,dep[x]-dep[top[x]],W[x]);
        }
        else if(op[0]=='C'&&op[1]=='W'){
            _scanf(x);_scanf(w);
            update(rt[top[x]][C[x]],0,len[top[x]]-1,dep[x]-dep[top[x]],W[x]=w);
        }
        else if(op[1]=='S'){
            _scanf(x);_scanf(y);
            z=LCA(x,y);
            _printf(query_s(x,z,C[x])+query_s(y,z,C[x])-(C[z]==C[x]?W[z]:0));
        }
        else if(op[1]=='M'){
            _scanf(x);_scanf(y);
            z=LCA(x,y);
            _printf(max(query_m(x,z,C[x]),query_m(y,z,C[x])));
        }
    }
}
int main()
{
    data_in();
    work();
    return 0;
}