思维+dfs+二染色,CF 1060E - Sergey and Subway

一、题目

1、题目描述

2、输入输出

2.1输入

2.2输出

3、原题链接

1060E - Sergey and Subway

---

二、解题报告

1、思路分析

考虑修改后的图，任意两点间的距离有何变化？

除2向上取整

那么我们要求的就是Σ[d / 2]

直接求太慢了，考虑求每个边的贡献，和奇数路径点对的贡献

每个边的贡献为上方点的数目乘下方点的数目，这个我们将权值下放到点上，dfs求子树大小即可

那么奇数路径点的数目如何求呢？

我们对树进行二染色，显然颜色不同的点对间路径权值就是奇数

那么问题就迎刃而解了

2、复杂度

时间复杂度： O(N)空间复杂度：O(N)

3、代码详解

 ​

#include <bits/stdc++.h>
using i64 = long long;
using i128 = __int128;
using PII = std::pair<int, int>;
using PIII = std::pair<int, PII>;
const int inf = 1e9 + 7, P = 1e9 + 7;

void solve() {
    int n;
    std::cin >> n;
    std::vector<std::vector<int>> g(n);
    for (int i = 1, a, b; i < n; ++ i) {
        std::cin >> a >> b;
        -- a, -- b;
        g[a].push_back(b);
        g[b].push_back(a);
    }


    std::vector<int> sz(n, 1);
    std::vector<bool> color(n);
    int con = 0;

    auto dfs = [&](auto&& self, int u, int fa) -> void {
        con += color[u];
        for (int v : g[u]) {
            if (v == fa) continue;
            color[v] = color[u] ^ 1;
            self(self, v, u);
            sz[u] += sz[v];
        }
    };

    dfs(dfs, 0, -1);

    i64 res = 1LL * con * (n - con);
    for (int x : sz)
        res += 1LL * (n - x) * x;
    std::cout << res / 2;
}


int main(int argc, char** argv) {
    std::ios::sync_with_stdio(false), std::cin.tie(0), std::cout.tie(0);
    int _ = 1;
    // std::cin >> _;
    while (_ --)
        solve();
    return 0;
}