[tyvj1520]树的直径（dfs||树形dp）

题目描述

传送门

题解

题目中就给了一种dfs的方法。
dp：f[i]g[i]分别表示i点到它的子树中的点距离的最大值和次大值。

代码

dfs

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int max_n=1e4+5;
const int max_e=max_n*2;
int n,x,y,z,ans,ansp;
int tot,point[max_n],next[max_e],v[max_e],c[max_e];
inline void add(int x,int y,int z){++tot;next[tot]=point[x];point[x]=tot;v[tot]=y;c[tot]=z;}
inline void dfs(int x,int fa,int dis){
    if (dis>ans) ans=dis,ansp=x;
    for (int i=point[x];i;i=next[i])
      if (v[i]!=fa)
        dfs(v[i],x,dis+c[i]);
}
inline void get_ans(int x,int fa,int dis){
    if (dis>ans) ans=dis,ansp=x;
    for (int i=point[x];i;i=next[i])
      if (v[i]!=fa)
        get_ans(v[i],x,dis+c[i]);
}
int main(){
    scanf("%d",&n);
    for (int i=1;i<n;++i) scanf("%d%d%d",&x,&y,&z),add(x,y,z),add(y,x,z);
    ans=0,ansp=1; dfs(1,0,0);
    ans=0; get_ans(ansp,0,0);
    printf("%d\n",ans);
}

dp

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int max_n=1e4+5;
const int max_e=max_n*2;
int n,x,y,z,ans;
int f[max_n],g[max_n];
int tot,point[max_n],next[max_e],v[max_e],c[max_e];
inline void add(int x,int y,int z){++tot;next[tot]=point[x];point[x]=tot;v[tot]=y;c[tot]=z;}
inline void treedp(int x,int fa){
    for (int i=point[x];i;i=next[i])
      if (v[i]!=fa){
        treedp(v[i],x);
        if (f[v[i]]+c[i]>f[x]){
            g[x]=f[x];
            f[x]=f[v[i]]+c[i];
        }
        else g[x]=max(g[x],f[v[i]]+c[i]);
      }
      ans=max(ans,f[x]+g[x]);
}
int main(){
    scanf("%d",&n);
    for (int i=1;i<n;++i) scanf("%d%d%d",&x,&y,&z),add(x,y,z),add(y,x,z);
    treedp(1,0);
    printf("%d\n",ans);
}