社区讨论

换根DP怎么会TLE的

P3478[POI 2008] STA-Station参与者 3已保存回复 3

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
3 条
当前快照
1 份
快照标识符
@mma4aocd
此快照首次捕获于
2026/03/03 12:38
上周
此快照最后确认于
2026/03/06 13:20
4 天前
查看原帖
CPP
#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,x,y,z,dp[1000001],sz[1000001],dep[1000001],ans=1;
struct edge{
	int v,w;
};
vector<edge> vec[1000001];
void dfs(int u,int fa){
	for(auto i:vec[u]){
		int v=i.v,w=i.w;
		if(v==fa) continue;
		dp[v]=dp[u]+w*(n-2*sz[v]);
		dfs(v,u);
	}
}
void dfs2(int u,int deep,int fa){
	sz[u]=1;
	dep[u]=deep;
	for(auto i:vec[u]){
		int v=i.v,w=i.w;
		if(v==fa) continue;
		dfs2(v,deep+w,u);
		sz[u]+=sz[v];
	}
}
signed main(){
	cin>>n;
	for(int i=1;i<n;i++){
		cin>>x>>y;
		vec[x].push_back({y,1});
		vec[y].push_back({x,1});
	}
	dfs2(1,0,0);
	for(int i=1;i<=n;i++) dp[1]+=dep[i];
	dfs(1,0);
	for(int i=1;i<=n;i++){
		if(dp[ans]<dp[i]) ans=i;
	}
	cout<<ans;
}
这不是O(n)吗

回复

3 条回复,欢迎继续交流。

正在加载回复...