换根DP模板

给你一个无根树,问你以哪个节点为根节点的时候得到所有点的深度之和最大

《贴一张 知乎大佬的一个解释》

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
const int N = 2e6+10;
using ll = long long;
ll dep[N],sz[N];
ll ans[N];
vector<int>g[N];
int n;

void dfs(int u,int father){
	sz[u] = 1;
	dep[u] = dep[father]+1;
	for(auto &t:g[u]){
		if(t==father)continue;
		dfs(t,u);
		sz[u]+=sz[t];
	}	
}


void down(int u,int father){
	
	for(auto &t:g[u]){
		if(t==father)continue;
		ans[t] = ans[u]-2*sz[t]+sz[1];
		down(t,u);
	}
}


int main()
{
	cin>>n;
	for(int i=1;i<n;i++){
		int a,b;cin>>a>>b;
		g[a].push_back(b);
		g[b].push_back(a);
	}
	
	dfs(1,0);
	for(int i=1;i<=n;i++)ans[1]+=dep[i];
	//for(int i=1;i<=n;i++)cout<<sz[i]<<"\n";
	down(1,-1);
	ll res = ans[1],p = 1;
	for(int i=1;i<=n;i++)if(ans[i]>res)res=ans[i],p=i;
	cout<<p;
}
相关推荐
张人玉2 小时前
C# 常量与变量
java·算法·c#
weixin_446122462 小时前
LinkedList剖析
算法
百年孤独_3 小时前
LeetCode 算法题解:链表与二叉树相关问题 打打卡
算法·leetcode·链表
我爱C编程4 小时前
基于拓扑结构检测的LDPC稀疏校验矩阵高阶环检测算法matlab仿真
算法·matlab·矩阵·ldpc·环检测
算法_小学生4 小时前
LeetCode 75. 颜色分类(荷兰国旗问题)
算法·leetcode·职场和发展
运器1234 小时前
【一起来学AI大模型】算法核心:数组/哈希表/树/排序/动态规划(LeetCode精练)
开发语言·人工智能·python·算法·ai·散列表·ai编程
算法_小学生4 小时前
LeetCode 287. 寻找重复数(不修改数组 + O(1) 空间)
数据结构·算法·leetcode
岁忧4 小时前
(LeetCode 每日一题) 1865. 找出和为指定值的下标对 (哈希表)
java·c++·算法·leetcode·go·散列表
alphaTao4 小时前
LeetCode 每日一题 2025/6/30-2025/7/6
算法·leetcode·职场和发展
ゞ 正在缓冲99%…4 小时前
leetcode67.二进制求和
算法·leetcode·位运算