CF 1230E Kamil and Making a Stream

给出一棵树, 点上有权值, 对于每个点, 求它和所有后代节点的GCD的和.

一开始的思路: 用当前节点更新祖先节点…然而对于每个点会收到n个点的贡献.
实际上, 如果我们用当前节点更新后代节点的贡献, 那么每个点只会收到log(n)个贡献, 这样就不会T了.

map<ll, int> mp[M];
vector<int> to[M];
ll x[M],ans = 0;
int n;
void dfs(int now, int fa) {
    for (auto a:mp[fa]) {
        ll tmp = __gcd(a.first, x[now]);
        mp[now][tmp] += a.second;
        ans = (ans + tmp % mod * a.second % mod) % mod;
    }
    mp[now][x[now]]++;
    ans = (ans + x[now]) % mod;;
    for (int son : to[now]) {
        if (son == fa)continue;
        dfs(son, now);
    }
}
int main() {
    n = read();
    for (int i = 1; i <= n; i++)x[i] = read();
    for (int i = 0; i < n - 1; i++) {
        int u = read(), v = read();
        to[u].emplace_back(v);
        to[v].emplace_back(u);
    }
    dfs(1, 0);
    write(ans), enter;
    return 0;
}