bzoj 3531: [Sdoi2014]旅行 线段树+树链剖分

题意

S国有N个城市，编号从1到N。城市间用N-1条双向道路连接，满足
从一个城市出发可以到达其它所有城市。每个城市信仰不同的宗教，如飞天面条神教、隐形独角兽教、绝地教都是常见的信仰。为了方便，我们用不同的正整数代表各种宗教， S国的居民常常旅行。旅行时他们总会走最短路，并且为了避免麻烦，只在信仰和他们相同的城市留宿。当然旅程的终点也是信仰与他相同的城市。S国政府为每个城市标定了不同的旅行评级，旅行者们常会记下途中（包括起点和终点）留宿过的城市的评级总和或最大值。
在S国的历史上常会发生以下几种事件：
”CC x c”：城市x的居民全体改信了c教；
”CW x w”：城市x的评级调整为w;
”QS x y”：一位旅行者从城市x出发，到城市y，并记下了途中留宿过的城市的评级总和；
”QM x y”：一位旅行者从城市x出发，到城市y，并记下了途中留宿过
的城市的评级最大值。
由于年代久远，旅行者记下的数字已经遗失了，但记录开始之前每座城市的信仰与评级，还有事件记录本身是完好的。请根据这些信息，还原旅行者记下的数字。 为了方便，我们认为事件之间的间隔足够长，以致在任意一次旅行中，所有城市的评级和信仰保持不变。
N，Q < =10^5 ， C < =10^5

分析

我真是太傻比了。。。
一开始想过建C棵线段树，但觉得空间分分钟爆掉，所以直接就被否决掉了，没想到还真的是这样。。。
只要动态开点一下就好了。

一次AC就当是练手速了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=100005;

int n,q,cnt,last[N],pos[N],top[N],fa[N],size[N],root[N],tot,sz,dep[N],c[N],w[N];
struct edge{int to,next;}e[N*2];
struct tree{int l,r,mx,s;}t[N*50];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;
}

void dfs1(int x)
{
    dep[x]=dep[fa[x]]+1;size[x]=1;
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa[x]) continue;
        fa[e[i].to]=x;
        dfs1(e[i].to);
        size[x]+=size[e[i].to];
    }
}

void dfs2(int x,int chain)
{
    pos[x]=++sz;top[x]=chain;int k=0;
    for (int i=last[x];i;i=e[i].next)
        if (e[i].to!=fa[x]&&size[e[i].to]>size[k]) k=e[i].to;
    if (!k) return;
    dfs2(k,chain);
    for (int i=last[x];i;i=e[i].next)
        if (e[i].to!=fa[x]&&e[i].to!=k) dfs2(e[i].to,e[i].to);
}

void updata(int d)
{
    t[d].mx=max(t[t[d].l].mx,t[t[d].r].mx);
    t[d].s=t[t[d].l].s+t[t[d].r].s;
}

void ins(int &d,int l,int r,int x,int y)
{
    if (!d) d=++tot;
    if (l==r)
    {
        t[d].s=t[d].mx=y;
        return;
    }
    int mid=(l+r)/2;
    if (x<=mid) ins(t[d].l,l,mid,x,y);
    else ins(t[d].r,mid+1,r,x,y);
    updata(d);
}

int finds(int d,int l,int r,int x,int y)
{
    if (!d||x>y) return 0;
    if (l==x&&r==y) return t[d].s;
    int mid=(l+r)/2;
    return finds(t[d].l,l,mid,x,min(y,mid))+finds(t[d].r,mid+1,r,max(x,mid+1),y);
}

int findmx(int d,int l,int r,int x,int y)
{
    if (!d||x>y) return 0;
    if (l==x&&r==y) return t[d].mx;
    int mid=(l+r)/2;
    return max(findmx(t[d].l,l,mid,x,min(y,mid)),findmx(t[d].r,mid+1,r,max(x,mid+1),y));
}

int querys(int x,int y)
{
    int ans=0,col=c[x];
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        ans+=finds(root[col],1,n,pos[top[x]],pos[x]);
        x=fa[top[x]];
    }
    if (dep[x]<dep[y]) swap(x,y);
    ans+=finds(root[col],1,n,pos[y],pos[x]);
    return ans;
}

int querymx(int x,int y)
{
    int ans=0,col=c[x];
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        ans=max(ans,findmx(root[col],1,n,pos[top[x]],pos[x]));
        x=fa[top[x]];
    }
    if (dep[x]<dep[y]) swap(x,y);
    ans=max(ans,findmx(root[col],1,n,pos[y],pos[x]));
    return ans;
}

int main()
{
    n=read();q=read();
    for (int i=1;i<=n;i++) w[i]=read(),c[i]=read();
    for (int i=1;i<n;i++)
    {
        int x=read(),y=read();
        addedge(x,y);
    }
    dfs1(1);
    dfs2(1,1);
    for (int i=1;i<=n;i++) ins(root[c[i]],1,n,pos[i],w[i]);
    while (q--)
    {
        char ch[3];
        scanf("%s",ch);
        int x=read(),y=read();
        if (ch[1]=='C')
        {
            ins(root[c[x]],1,n,pos[x],0);
            c[x]=y;
            ins(root[c[x]],1,n,pos[x],w[x]);
        }
        else if (ch[1]=='W')
        {
            w[x]=y;
            ins(root[c[x]],1,n,pos[x],w[x]);
        }
        else if (ch[1]=='S') printf("%d\n",querys(x,y));
        else printf("%d\n",querymx(x,y));
    }
    return 0;
}