求助简化代码
查看原帖
求助简化代码
927266
bbbbbbbbbbbbbbbbbbb楼主2025/2/2 22:06
#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,m;
const int N=2e5+7;
struct node{
	int v,w;
};
vector<node> G[N];
int far,sum1[N],sum2[N];
void dfs1(int u,int fa){
	if(sum1[u]>sum1[far]) far=u;
	for(node l:G[u]){
		if(l.v==fa) continue;
		sum1[l.v]=sum1[u]+l.w;
		dfs1(l.v,u);
	}
}
void dfs2(int u,int fa){
	for(node l:G[u]){
		if(l.v==fa) continue;
		sum2[l.v]=sum2[u]+l.w;
		dfs2(l.v,u);
	}
}
signed main(){
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin>>n>>m;
	for(int i=1,u,v,w;i<=m;i++){
		cin>>u>>v>>w;
		G[u].push_back({v,w}),G[v].push_back({u,w});
	}
	dfs1(1,0);int s=far;
	sum1[s]=0;dfs1(far,0);int e=far;
	dfs2(e,0);
	int ans=0;
	for(int i=1;i<=n;i++){
		ans=max(ans,sum1[e]+min(sum1[i],sum2[i]));
	}
	cout<<ans<<'\n';
	return 0;
}

sum1 与 sum2 代表直径端点与树中各个点之间的距离,dfs1 与 dfs2 的流程差不多,能不能用一个指针数组取 sum1 与 sum2 的地址,简化为只写一个 dfs。但是本蒟蒻不会写,请大佬指教

2025/2/2 22:06
加载中...