luoguP2664 树上游戏（点分治）

前言：
去年舒老师还健在的时候，这是ta问我的最后一道题
为了纪念ta，我打算在这个神圣的日子里完成一下这道题

想当年，我看到这道题时一秒口hu：树上莫队
（然而真的是树上莫队的裸题）
在不学新知识的情况下，试着用点分治搞一下

题目链接

分析：
点分治——以我现在水平——认为只能解决子树内的路径计数之类的问题
要解决路径问题还是第一次见
显然我们应该是计算以每一个节点为LCA的点对之间路径的贡献

以舒老师的说法：
很明显，这种处理树上点对、nlogn可以接受且单次询问的问题，肯定是点分治跑不掉了

我们现在的问题就是：如何计算以某个结点为LCA的点对贡献

通过大量的画图和发呆
我发现了几个性质：

• 对于子树x，若其中i结点的颜色是独一无二的，那么i就可以与所有非该子树内的点形成点对，贡献就是size[root]-size[x]
• 对于树中的一点ｉ，如果该点的颜色在该点到根这条链上是第一次出现，
  那么对于这棵树的其他点ｊ（j与i的lca为根结点），均能与i的子树（包括i）组成点对，ｉ的颜色会对j的答案贡献size[i]（我们暂且不考虑j到根的链上是否出现了i的颜色）

Q.

一开始我不是很理解第二个性质（这是舒老师传授的）
主要的疑惑就在于：要是i的子树内也有i的颜色怎么办呢

A.

但是注意原命题中有一个很重要的条件：ｉ的颜色在该点到根这条链上是第一次出现

我们在继续dfs时候，如果又遇到了重复颜色是不会计算贡献的，所以这个性质是很优秀的

这道题中两个性质都有应用

---

整体的思路当然还是dfs分治

void solve(int now,int fa)
{
    vis[now]=1;
    doit(now,fa);
    for (int i=st[now];i!=-1;i=way[i].nxt)
    {
        int v=way[i].y;
        if (v!=fa&&!vis[v])
        {
            sz=size[v];
            root=0; 
            getroot(v,0);
            solve(root,0); 
        }
    }
}

然而在分治的时候计算贡献就有一点点麻烦了：

dfs_1

什么都别啰嗦，一上来先dfs一遍
维护每个结点的子树大小
（因为每次我们solve的时候，root都有可能变化，所以每次都要维护size）
同时计算第二个性质中牵扯的贡献

这个贡献要怎么维护呢？
因为贡献牵扯到size，所以我们只能在回溯的时候记录
设置一个计数器cnt，按照dfs顺序进行对应颜色的计数
在回溯的时候，如果cnt=1，那么这个颜色对于当前这棵树的贡献就是size[i]
我们把每种颜色的贡献记录在一个数组中（color）
最后，因为我们要向上回溯，所以不要忘了消除影响

if (cnt[C[now]]==1)
{
    sum+=size[now];
    color[C[now]]+=size[now];     //C[now]对于当前这棵树的贡献
} 

我们可以注意到代码中有一个变量：$sum$
这个就是子树内每种颜色的贡献总和

枚举子树

void doit(int now,int fa)
{
    dfs_1(now,fa);
    ans[now]+=(ll)sum;
    for (int i=st[now];i!=-1;i=way[i].nxt)
    {
        int v=way[i].y;
        if (!vis[v]&&v!=fa)
        {
            cnt[C[now]]++;
            color[C[now]]-=size[v];  sum-=size[v];
            change(v,now,-1);
            cnt[C[now]]--;

            tt=size[now]-size[v];
            dfs_2(v,now);

            cnt[C[now]]++;                   //消除影响 
            color[C[now]]+=size[v]; sum+=size[v];
            change(v,now,1);
            cnt[C[now]]--;
        }
    }
    sum=0;num=0;
    clear(now,fa);
}

我们这样只处理了：以根结点为LCA的点对贡献，且两点一定在该子树内
但是对于那些“以根结点为LCA，但只有一个点在该子树内”的点对，我们没有计算，
而第二类点对对于根结点是不存在的（都已经是根结点了，不存在不在ta管辖下的点）
我们要考虑的是根结点的儿子，并且给这些点维护第二类贡献

枚举所有子树，先把子树扫一遍清除其在color数组中的所有贡献（change函数）
（因为根结点的颜色一定是“在该点到根这条链上是第一次出现”，所以我们要先 $-size[son]$ ）

void change(int now,int fa,int z)
{
    cnt[C[now]]++;
    for (int i=st[now];i!=-1;i=way[i].nxt)
    {
        int v=way[i].y;
        if (v!=fa&&!vis[v]) change(v,now,z);
    }
    if (cnt[C[now]]==1)
    {
        sum+=(ll)size[now]*z;
        color[C[now]]+=(ll)size[now]*z;
    }
    cnt[C[now]]--;
}

记 $tt=size[root]-size[该子树]$
这样我们就进入了下一个函数：

dfs_2

void dfs_2(int now,int fa)
{
    cnt[C[now]]++;
    if (cnt[C[now]]==1)
    {
        sum-=color[C[now]];
        num++;
    }
    ans[now]+=(ll)sum+num*tt;
    for (int i=st[now];i!=-1;i=way[i].nxt)
    {
        int v=way[i].y;
        if (v!=fa&&!vis[v]) dfs_2(v,now);
    }
    if (cnt[C[now]]==1)
    {
        sum+=color[C[now]];
        num--;
    }
    cnt[C[now]]--;
}

这里我们就需要利用第一条性质了

我们遍历整棵子树
对于该子树中的每一个点ｊ：
设Ｘ＝Σcolor[j 到根上（不包括根）的所有颜色]
由于这些颜色在子树中出现过，我们可以通过之后的分治记录ta们的贡献，所以这里就不用计算了
设$num$为j到根上（不包括根）的颜色数

则ans[j]+＝sum-X+num*tt

clear

void clear(int now,int fa)
{
    cnt[C[now]]=0;
    color[C[now]]=0;
    for (int i=st[now];i!=-1;i=way[i].nxt)
    {
        int v=way[i].y;
        if (v!=fa&&!vis[v]) clear(v,now);
    }
}

清空cnt和color数组，方便下一步的分治