2019牛客国庆集训派对day4H题

题目链接：https://ac.nowcoder.com/acm/contest/1109/H

题意：

给一棵带边权的树，现在要求新建一棵树，新树里(u,v)的边权等于原树中(u,v)的唯一路径的距离。
现在要你找出最大的花费来建造这颗新树。

题解：

很容易知道这个题的求出这个树的两个直径端点，其他每个点到这两个端点的最大距离和就是答案。

之后就是求该树的直径端点，关于这个算法有两种经典的做法，bfs和树形dp，

这里用的是三次dfs，写起来比较简单，在dfs求两个端点的同时更新每个点到端点的距离就行了。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
const ll INF=0x3f3f3f3f3f3f3f3fll;
vector<pair<int,int> >G[maxn];
int visited[maxn];
ll dist=-INF;
int point;
ll res[3][maxn];
int n;
void dfs(int x,ll dis,int index)
{
    if(index)
		res[index][x]=dis;
    if(dis>dist){
    	dist=dis;
		point=x;
	}
	int size=G[x].size();
    for(int i=0;i<size;i++){
        if(visited[G[x][i].first]) continue;
        visited[G[x][i].first]=1;
        dfs(G[x][i].first,G[x][i].second+dis,index);
    }
}
void solve()
{ 
    memset(visited,0,sizeof(visited));
    dist=-INF;
    dfs(1,0,0);
    int x=point;
    
    memset(visited,0,sizeof(visited));
    dist=-INF;
    dfs(x,0,1);
    int y=point;
    
    memset(visited,0,sizeof(visited));
    dfs(y,0,2);
    
    ll ans=res[1][y];
    for(int i=0;i<=n;i++){
        if(i==x||i==y) continue;
        ans+=max(res[1][i],res[2][i]);
    }
    printf("%lld\n",ans);
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=0;i<maxn;i++) G[i].clear();
        for(int i=0;i<n-1;i++)
        {
            int u,v;
			ll w;
            scanf("%d%d%d",&u,&v,&w);
            G[u].push_back(make_pair(v,w));
            G[v].push_back(make_pair(u,w));    
        }
        solve();
    }
    return 0;
}