AC日记——Two poj 1849

Two

思路：

　　树形DP求直径；

　　答案是边权总和*2-直径；

　　dp[i][1]::以i为根的子树中最长的路径；

　　dp[i][0]::以i为根的子树中次长的路径；

来，上代码：

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

using namespace std;

#define maxn 100005

int n,s,E[maxn<<],V[maxn<<],W[maxn<<],cnt;
int head[maxn],dp[maxn][],sum,ans=;

inline void in(int &now)
{
    char Cget=getchar();now=;
    while(Cget>''||Cget<'') Cget=getchar();
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
}

void dfs(int now,int fa)
{
    bool res=true;
    int pos=,gs=;
    for(int i=head[now];i;i=E[i])
    {
        if(V[i]==fa) continue;
        res=false,dfs(V[i],now);
        if(W[i]+dp[V[i]][]>gs) gs=W[i]+dp[V[i]][],pos=V[i];
    }
    if(res) return ;
    dp[now][]=gs,gs=;
    for(int i=head[now];i;i=E[i])
    {
        if(V[i]==fa||V[i]==pos) continue;
        if(W[i]+dp[V[i]][]>gs) gs=W[i]+dp[V[i]][];
    }
    dp[now][]=gs,ans=max(ans,dp[now][]+dp[now][]);
}

int main()
{
    in(n),in(s);int u,v,w;
    for(int i=;i<n;i++)
    {
        in(u),in(v),in(w),sum+=w*;
        E[++cnt]=head[u],V[cnt]=v,W[cnt]=w,head[u]=cnt;
        E[++cnt]=head[v],V[cnt]=u,W[cnt]=w,head[v]=cnt;
    }
    dfs(s,);
    cout<<sum-ans;
    return ;
}