换根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;
}
相关推荐
yi.Ist27 分钟前
数据结构 —— 键值对 map
数据结构·算法
s1533530 分钟前
数据结构-顺序表-猜数字
数据结构·算法·leetcode
Coding小公仔33 分钟前
LeetCode 8. 字符串转换整数 (atoi)
算法·leetcode·职场和发展
GEEK零零七39 分钟前
Leetcode 393. UTF-8 编码验证
算法·leetcode·职场和发展·二进制运算
DoraBigHead2 小时前
小哆啦解题记——异位词界的社交网络
算法
木头左3 小时前
逻辑回归的Python实现与优化
python·算法·逻辑回归
lifallen7 小时前
Paimon LSM Tree Compaction 策略
java·大数据·数据结构·数据库·算法·lsm-tree
web_Hsir9 小时前
vue3.2 前端动态分页算法
前端·算法
地平线开发者11 小时前
征程 6M 部署 Omnidet 感知模型
算法·自动驾驶
秋说12 小时前
【PTA数据结构 | C语言版】线性表循环右移
c语言·数据结构·算法