洛谷U81904 【模板】树的直径

有负边权，所以用树形DP来找树的直径。

//树形DP求树的直径 
#include<bits/stdc++.h>
using namespace std;
const int N=500005,M=500005;
int n,m,tot,ans;
int f1[N],f2[N];//以u为根的子树的最大值和次大值 
int head[N],to[M*2],w[M*2],nxt[M*2];

void add(int x,int y,int z){
    nxt[++tot]=head[x];
    head[x]=tot;
    to[tot]=y;
    w[tot]=z;
}

void dp(int u,int fa){
    for(int i=head[u];i;i=nxt[i]){
        int v=to[i];
        if(v==fa) continue;
        dp(v,u);
        if(f1[u]<f1[v]+w[i]){
            f2[u]=f1[u];
            f1[u]=f1[v]+w[i];
        }
        else if(f2[u]<f1[v]+w[i])
            f2[u]=f1[v]+w[i];
        ans=max(ans,f1[u]+f2[u]);
    }
}

int main(){
    scanf("%d",&n);
    int x,y,z;
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);add(y,x,z);
    }
    dp(1,0);
    printf("%d\n",ans);
    return 0;
}

 

---

如果您觉得阅读本文对您有帮助，请点一下“推荐”按钮，您的“推荐”将是我最大的写作动力！欢迎各位转载，但是未经作者本人同意，转载文章之后必须在文章页面明显位置给出作者和原文连接，否则保留追究法律责任的权利。