链接:https://ac.nowcoder.com/acm/contest/11371/B
来源:牛客网
题目描述
有一天clccle和rqy走在某个国家的街头上,机智的rqy却发现周围的行人不太对劲,他们嘴里念念有词,说着"sqn tql!",一边漫无目的的行走,clccle也发现了这一点,却惊讶的发觉这种奇怪的病毒会向周围的城市,最终会感染整个国家,因为网络已经崩溃,所以她们忘记了自己所在的城市,她们唯一知道的是这种病毒是从当前她们所在的城市开始传播的,并且这个国家的所有城市到这个城市的距离和最小(所有道路的距离都为1),现在给定聪明的你一张整个国家的地图,请你帮rqy和clccle找到她们现在可能在这个国家的哪一个城市.
输入描述:
两个整数n,m,代表这个国家一共有n个城市,城市之间只有m条道路
接下来m行,每行两个整数a,b代表城市a,b之间有一条联通的道路
输出描述:
多个整数,输出当前clccle和rqy可能所在的点
示例1
输入
复制
2 1
1 2
输出
复制
1 2
备注:
对于所有的数据,1<=m<=n<=50000
(图的类型保证没有大小大于等于3的环)
树形dp求出每个点到所有点的距离和,树形dp即可,关键是讨论出来父亲
对孩子的贡献,和孩子对父亲的贡献
第一次dfs的时候从下向上传递 , 此时dp[u]表示u点到子树中所有
点的最短距离之和 , x节点对当前节点的贡献是 dp[x] + size[x] , 其
含义在孩子节点x的基础上面,所有的子节点全部深度全部加一 ,
也就是 + size[x]
从上到下传递的时候,当前节点u对孩子节点的贡献是 dp[u] - dp[x]
- size[x] + size[u] - size[x] ,解释: 先将当前节点的dp[u] 剪掉 所
- 要传孩子节点x的dp[x] 贡献删掉, 根据第一步 , 其贡献是dp[x]
- + size[x] , size[u] 要减掉size[x] , 这两个dp[u] - dp[x] - size[x] , 和
- size[u] - size[x] , 都相当于在下传贡献的时候, 先将孩子节点的
- 贡献删掉 , 然后将孩子节点数量改变一下,因为下一次x节点就变
- 成了根节点 , dp值要变, size孩子节点也要变, 之后要用到
一般来说,树形dp维护节点的尺寸是非常必要的,如果讨论子节点对父亲节点的贡献的话,一般是先递归求出子节点的信息,再求子节点对父亲节点的贡献,求父亲节点对子节点的贡献的时候是先维护出父亲节点的信息,再去求子节点的信息
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <set>
#include <string>
#include <queue>
#include <map>
#include <stack>
#include <map>
#include <unordered_map>
#include <vector>
#include <cmath>
#include <ext/rope>
#include <bits/stdc++.h>
using namespace std;
#define gt(x) x = read()
#define int long long
#define ios ios::sync_with_stdio(false) , cin.tie(0) , cout.tie(0)
const int mod = 1e9 + 7;
inline int read(int out = 0)
{
char c;
while((c=getchar()) < 48 || c > 57);
while(c >= 48 && c <= 57) out=out*10+c-48,c=getchar();
return out;
}
const int N = 500010;
const int M = 1e6 + 10;
int n, m;
int h[N], e[N], ne[N], idx;
int f[N], size1[N];
void add(int a, int b){
e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}
void dfs(int u, int father){
size1[u] = 1;
for (int i = h[u]; ~i; i = ne[i]){
int j = e[i];
if (j == father) continue;
dfs(j, u);
size1[u] += size1[j];
f[u] += (f[j] + size1[j]);
}
}
void dfs1(int u, int father, int sum, int res){
size1[u] += res;
f[u] += sum;
for (int i = h[u]; ~i; i = ne[i]){
int j = e[i];
if (j == father) continue;
dfs1(j, u, f[u] - f[j] - 2 * size1[j] + size1[u], size1[u] - size1[j]);
}
}
signed main(){
ios;
gt(n), gt(m);
memset(h, -1, sizeof h);
while(m --){
int a, b;
gt(a), gt(b);
add(a, b), add(b, a);
}
dfs(1, -1);
dfs1(1, -1, 0, 0);
int ans = 0x7f7f7f7f;
for (int i = 1; i <= n; i ++){
ans = min(ans, f[i]);
}
for (int i = 1; i <= n; i ++){
if (f[i] == ans){
cout << i << " ";
}
}
cout << endl;
return 0;
}