一、题目
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;
}