AcWing 1072. 树的最长路径

题目链接: https://www.acwing.com/problem/content/1074/

听y总的课听睡着了, 直接看的题解琢磨了下, 感觉挺简单的

题意

给你一棵树, 找到这棵树的最长路径

题解

先说屁话, 感觉能用最长路写, 最短路改改就好了
当然这是个树形dp的模板题这里讲解树形dp的做法
与其说是树形dp, 我觉得更像是dfs遍历了一遍树而已, 感觉没什么状态转移的特点在里面

直接讲dfs的过程, 存图用前向星, 前向星的基础知识就不讲了
因为是树的最长路径, 所以中间不能断开(废话)
所以对于每个子节点, 必须和他的父亲节点相连, 我们要找的是最长路径, 从根节点开始搜的话需要找到距离根节点最长和次长的路径
在每次dfs时, 遍历此节点的所有儿子, 获取其儿子到叶子结点的最长路径, 每次遍历都更新维护一个最长路径和次长路径
当dfs回到根节点时, 此时维护的最长路径和次长路径就是叶子结点到达根节点的路径
整理一下输出最大值(是最长路径和次长路径的最大值)即可
注意最后一个最长路径和次长路径的和不一定是最大的
因为最长路径不一定会经过根节点!!!

ll cnt,n,m,t,ans,ant;
const int N=1e4+10;
const int INF=0x3f3f3f3f;
const ll llINF=0x3f3f3f3f3f3f3f3f;
ll h[N],e[N*2],w[N*2],ne[N*2],idx;
string str;

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

ll dfs(ll u,ll fa)//当前节点和父节点
{
    ll dist=0;
    ll d1=0,d2=0;//最长路和次长路

    for(ll i=h[u];i!=-1;i=ne[i])
    {
        ll j=e[i];
        if(j==fa) continue;
        ll d=dfs(j,u)+w[i];//路径总长度
        dist=max(dist,d);//路径最大值
        if(d>=d1) 
        {
            d2=d1;
            d1=d;
        }else d2=max(d,d2);
    }

    ans=max(ans,d1+d2);
    return dist;
}

void solve()
{
    memset(h,-1,sizeof h);
    cin>>n;
    repr(i,1,n)
    {
        ll a,b,w;
        cin>>a>>b>>w;
        add(a,b,w);
        add(b,a,w);//无向边
    }
    
    dfs(1,-1);
    cout<<ans<<endl;
    
    return;
}