cf1172E Nauuo and ODT（LCT）

首先可以转化问题，变为对每种颜色分别考虑不含该颜色的简单路径条数。然后把不是当前颜色的点视为白色，是当前颜色的点视为黑色，显然路径数量是每个白色连通块大小的平方和，然后题目变为：黑白两色的树，单点翻转颜色，维护白色连通块大小平方和，然后根据Auuan大佬的题解，我用了LCT。就是对每个点维护子树、儿子大小平方和，在 link/cut 的时候更新答案。初始化所有点是白色，离线处理每个颜色即可。

这题放在2h比赛上，除了lxl其他人都写不出来（况且lxl还是本题出题人呢）

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=4e5+;
int n,m,c[N],f[N],fa[N],ch[N][],sum[N],sz[N];
ll ans,d[N],sz2[N];
bool vis[N];
vector<int>vec[N][],G[N];
bool nroot(int x){return x==ch[fa[x]][]||x==ch[fa[x]][];}
void pushup(int x){sum[x]=sum[ch[x][]]+sum[ch[x][]]+sz[x]+;}
void rotate(int x)
{
    int y=fa[x],z=fa[y],w=x==ch[y][];
    if(nroot(y))ch[z][y==ch[z][]]=x;
    fa[x]=z,ch[y][w]=ch[x][w^],fa[ch[x][w^]]=y,ch[x][w^]=y,fa[y]=x;
    pushup(y),pushup(x);
}
void splay(int x)
{
    while(nroot(x))
    {
        int y=fa[x],z=fa[y];
        if(nroot(y))
        {
            if((x==ch[y][])^(y==ch[z][]))rotate(x);
            else rotate(y);
        }
        rotate(x);
    }
}
void access(int x)
{
    int y=;
    while(x)
    {
        splay(x);
        sz[x]+=sum[ch[x][]]-sum[y];
        sz2[x]+=1ll*sum[ch[x][]]*sum[ch[x][]]-1ll*sum[y]*sum[y];
        ch[x][]=y;
        pushup(x);
        x=fa[y=x];
    }
}
int findrt(int x)
{
    access(x),splay(x);
    while(ch[x][])x=ch[x][];
    splay(x);
    return x;
}
void link(int x)
{
    int y=f[x],z;
    splay(x);
    ans-=sz2[x]+1ll*sum[ch[x][]]*sum[ch[x][]];
    z=findrt(y);
    access(x),splay(z);
    ans-=1ll*sum[ch[z][]]*sum[ch[z][]];
    fa[x]=y;
    splay(y);
    sz[y]+=sum[x],sz2[y]+=1ll*sum[x]*sum[x];
    pushup(y),access(x),splay(z);
    ans+=1ll*sum[ch[z][]]*sum[ch[z][]];
}
void cut(int x)
{
    int y=f[x],z;
    access(x);
    ans+=sz2[x];
    z=findrt(y);
    access(x),splay(z);
    ans-=1ll*sum[ch[z][]]*sum[ch[z][]];
    splay(x);
    ch[x][]=fa[ch[x][]]=;
    pushup(x),splay(z);
    ans+=1ll*sum[ch[z][]]*sum[ch[z][]];
}
void dfs(int u)
{for(int i=;i<G[u].size();i++)if(G[u][i]!=f[u])f[G[u][i]]=u,dfs(G[u][i]);}
int main()
{
    scanf("%d%d",&n,&m);
    ll lst;
    for(int i=;i<=n;i++)
    scanf("%d",&c[i]),vec[c[i]][].push_back(i),vec[c[i]][].push_back();
    for(int i=;i<=n+;i++)sum[i]=;
    for(int i=,x,y;i<n;i++)scanf("%d%d",&x,&y),G[x].push_back(y),G[y].push_back(x);
    for(int i=,u,v;i<=m;i++)
    {
        scanf("%d%d",&u,&v);
        vec[c[u]][].push_back(u),vec[c[u]][].push_back(i);
        c[u]=v;
        vec[v][].push_back(u),vec[v][].push_back(i);
    }
    f[]=n+;
    dfs();
    for(int i=;i<=n;i++)link(i);
    for(int i=;i<=n;i++)
    {
        if(!vec[i][].size()){d[]+=1ll*n*n;continue;}
        if(vec[i][][])d[]+=1ll*n*n,lst=1ll*n*n;else lst=;
        for(int j=;j<vec[i][].size();j++)
        {
            int u=vec[i][][j];
            if(vis[u]^=)cut(u);else link(u);
            if(j==vec[i][].size()-||vec[i][][j+]!=vec[i][][j])
            d[vec[i][][j]]+=ans-lst,lst=ans;
        }
        for(int j=vec[i][].size()-;~j;j--)
        {
            int u=vec[i][][j];
            if(vis[u]^=)cut(u);else link(u);
        }
    }
    ans=1ll*n*n*n;
    for(int i=;i<=m;i++)ans-=d[i],printf("%lld\n",ans);
}