P2664题解 【树上游戏】

原题链接
被点分治虐的心态爆炸了

题解在此

发现直接统计路径上的颜色数量很难，考虑转化一下统计方式。对于某一种颜色cc，它对一个点的贡献为从这个点出发且包含这种颜色的路径条数。
于是我们先点分一下，然后分别统计经过分治中心的路径对根和对其他点的贡献就行了。
推荐一篇比较详细的题解
代码：

#include <bits/stdc++.h>

using namespace std;

#define N 100000
#define pb push_back
#define ll long long

int n, c[N + 5];
vector<int> G[N + 5];
int root, S, sz[N + 5], vis[N + 5], maxsz[N + 5], col[N + 5], w[N + 5];
ll cnt[N + 5], ans[N + 5], sum1, sum2;

void getRoot(int u, int pa) {
    sz[u] = 1; maxsz[u] = 0;
    for (auto v : G[u]) {
        if (v == pa || vis[v]) continue;
        getRoot(v, u);
        sz[u] += sz[v];
        maxsz[u] = max(maxsz[u], sz[v]);
    }
    maxsz[u