【非原创】codeforces 1060E Sergey and Subway 【树上任意两点距离和】
2024-10-16 20:32:30
学习博客:戳这里
本人代码:
1 #include <bits/stdc++.h>
2 using namespace std;
3 typedef long long ll;
4 const int maxn = 2e5 + 10;
5 const ll mod = 998244353;
6 vector<int> mp[maxn];
7 ll ans = 0, cnt[3];
8 int n;
9 ll dfs(int u, int pre,int now) {
10 ++cnt[now];
11 ll siz = 1;
12 for(int i = 0; i < mp[u].size(); ++i) {
13 int v = mp[u][i];
14 if(v == pre) continue;
15 siz += dfs(v, u, now^1);
16 }
17 ans += siz * (n - siz);
18 return siz;
19 }
20 int main() {
21
22
23 scanf("%d", &n);
24 int u, v;
25 for(int i = 1; i < n; ++i) {
26 scanf("%d %d", &u, &v);
27 mp[u].push_back(v);
28 mp[v].push_back(u);
29 }
30 dfs(1,0,0);
31 printf("%lld\n", (ans + cnt[0] * cnt[1]) / 2);
32 return 0;
33 }
最新文章
- 【JAVA】 @override报错的解决方法
- Java集合---LinkedList源码解析
- [转]初探 PhoneGap 框架在 Android 上的表现
- 主题: 为kindsoft编辑器替换SyntaxHighlighter代码高亮,整合DEDECMS
- java rest版简单的webservice
- SQL Server DBA工作内容详解
- CSS常见BUG
- RAID技术介绍和总结
- 机器学习:Python中如何使用最小二乘法
- [转] Web 开发模式演变历史和趋势
- C#反射调用方法实例
- 深入浅出KNN算法(一) KNN算法原理
- 通过语法设置DNS解析
- 2019.03.29 NOIP训练 友好国度(点分治+容斥)
- Apache-Axis小结
- Python数据分析--Pandas知识点(一)
- word自动导入目录
- 25-删除m位数是剩下的最大
- Laravel路由除了根目录全报404错误
- HDU4745——Two Rabbits——2013 ACM/ICPC Asia Regional Hangzhou Online