树的三个经典问题----树的重心、中心、最长路径

最新推荐文章于 2024-07-25 14:00:47 发布

原创

最新推荐文章于 2024-07-25 14:00:47 发布 · 449 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #算法

本文介绍如何求解树的最长路径、树的中心及树的重心等问题，并提供了具体的算法实现代码。针对不同情况（如权值为非负或包含负数），采用不同的算法策略。

1.树的最长路径

在这里插入图片描述

权值为非负时

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 10010;
int n;
int h[N],e[N*2],w[N*2],ne[N*2],idx;
int dist[N];
void add(int a,int b,int c){
   
   
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u,int father,int distance){
   
   
    dist[u]=distance;
    for(int i=h[u];i!=-1;i=ne[i]){
   
   
        int j=e[i];
        if(j!=father)
        dfs(j,u,distance+w[i]);
    }
}
int main()
{
   
   
    cin>>n;
    memset(h,-1,sizeof h);
    for(int i=0;i<n-1;i++){
   
   
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
        add(b,a,c);
    }
    dfs(1,-1,0);
    int u=1;
    for(int i=2;i<=n;i++){
   
   
        if(dist[u]<dist