[Acwing] 树的直径 树DP|DFS

前言

之前学LCA就学过了 知道了DFS方法

这次为了了解树DP,滚过来复习

思路

DFS :
我们只需要 从根节点开始跑到能跑到的最远点,再从该点枚举到最远点那么最后一次便是我们的直径

树形DP :
我们记录当 $1$为树的根时，每个节点作为子树的根向下，所能延伸的最远距离$d1$，和次远距离 $d2$，那么直径就是所有 $d1+d2$的最大值。

CODE

DFS

#include <bits/stdc++.h>
using namespace std;
const int N = 10000 + 10;

int n, c, d[N];
int h[N],w[N*2],e[N*2],ne[N*2],idx;


void add(int a,int b,int c)
{
    e[idx] = b,ne[idx] = h[a],w[idx]  =c,h[a]=idx++;
}

void dfs(int u, int fa)
{

    for(int i = h[u];~i;i=ne[i])
    {
        int v = e[i];
        if (v == fa) continue;
        d[v] = d[u] + w[i];

        if (d[v] > d[c]) c = v;
        dfs(v, u);
    }
}
void solve()
{
    cin>>n;
    memset(h,-1,sizeof h);

    for(int i=1; i<n; i++)
    {
        int u,v,c;
        cin>>u>>v>>c;
        add(u,v,c);
        add(v,u,c);
    }

    dfs(1,0);
    d[c] = 0 ;
    dfs(c,0);

    cout<<d[c]<<endl;
    return;
}

int main()
{
    ios::sync_with_stdio(false);
    solve();
    return 0;
}

树形DP :

#include <bits/stdc++.h>
using namespace std;
const int N = 10000 + 10;

int n, c, d1[N],d2[N];
int h[N],w[N*2],e[N*2],ne[N*2],idx;
int d;

void add(int a,int b,int c)
{
    e[idx] = b,ne[idx] = h[a],w[idx]  =c,h[a]=idx++;
}

void dfs(int u, int fa)
{
    d1[u] = d2[u] = 0 ;
    for(int i = h[u];~i;i=ne[i])
    {
        int v = e[i];
        if (v == fa) continue;
        dfs(v, u);

        int t = d1[v] + w[i];
        if(t>d1[u])
            d2[u]  =d1[u],d1[u] = t;
        else if(t>d2[u])
            d2[u] = t;
        //if (d[v] > d[c]) c = v;

    }
    d =max(d,d1[u]+d2[u]);
}
void solve()
{
    cin>>n;
    memset(h,-1,sizeof h);

    for(int i=1; i<n; i++)
    {
        int u,v,c;
        cin>>u>>v>>c;
        add(u,v,c);
        add(v,u,c);
    }
    dfs(1,0);
    cout<<d<<endl;

    return;
}

int main()
{
    ios::sync_with_stdio(false);
    solve();
    return 0;
}