CF 1230E Kamil and Making a Stream

最新推荐文章于 2021-07-20 20:23:52 发布

swqeaaa

最新推荐文章于 2021-07-20 20:23:52 发布

阅读量461

点赞数

本文链接：https://blog.csdn.net/qq_32259423/article/details/101303892

版权

本文探讨了一种针对树形结构中点权值的优化算法，旨在高效求解每个节点与其所有后代节点权值的最大公约数（GCD）之和。传统方法对于每个节点会收到n个点的贡献，导致效率低下。通过创新地使用当前节点更新后代节点的贡献，实现了每个节点仅接收log(n)级别的贡献，显著提升了算法效率。

摘要由CSDN通过智能技术生成

给出一棵树, 点上有权值, 对于每个点, 求它和所有后代节点的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;
}