本题的数据是 n <= 1000000 , 应该是O(n) 算法;
先考虑最朴素的算法 , 进行n 次的dfs 求出深度和 , 复杂度是O(n ^2);
如果我们能在第一次dfs的基础上在对其他点的进行O(1)的求解那就满足复杂度的要求了;
然后我们画一下图(我不太会画图就不画了);
假设我们已经知道了 以x为根的答案 , 然后求出它的子节点y的答案;
然后我们将根往y 移动 , 发现 原先以x为根的子树中 , 除去以y为根的子树 , 他们的深度都会加一 , 而 原先y的子树的所有节点都会减去1
所以答案asn_y = ans_x + size_x - size_y; (如下图)
这里附上我的代码
#include<cstdio> #include<queue> #include<map> #include<cstring> #include<cstdlib> #include<deque> #include<iostream> #include<vector> #include<algorithm> #define inf 0x3f3f3f3f #define M 1000010 using namespace std; int cnt , head[M] , next[M << 1] , u[M << 1] , n , ans, size[M]; long long d[M] , f[M] , maxn ; void add(int x , int y){ u[++cnt] = y , next[cnt] = head[x] , head[x] = cnt; } inline void dfs1(int x , int fa , int dep){ int y ; for (int i = head[x] ; i ; i = next[i]){ y = u[i]; if ( y == fa) continue; dfs1(y , x , dep + 1); size[x] += size[y] , d[x] += d[y]; } size[x] += 1 , d[x] += dep; } inline void dfs2(int x , int fa){ int y; for (int i = head[x] ; i ; i = next[i]){ y = u[i]; if (y == fa) continue; f[y] = f[x] + n - size[y] * 2; dfs2(y , x); } } int main(){ // freopen("A.out" , "w" , stdout); // freopen("c1.in" , "r" , stdin); scanf("%d" , &n); for (int i = 1 ; i < n ; ++i){ int a , b; scanf("%d%d" , &a , &b); add(a , b); add(b , a); } dfs1(1 , 0 , 0); f[1] = d[1]; dfs2(1 , 0 ); for (int i = 1 ; i <= n ; ++i) if (f[i] > maxn) ans = i , maxn = f[i]; printf ("%d" , ans); return 0; }