蓝桥杯 PREV-9 大臣的旅费 （两次DFS）

题目链接：http://lx.lanqiao.cn/problem.page?gpid=T32

题意：求树中最远的两个点的距离。

解题方案：第一遍dfs，任取一个点，找到离它最远的点a，第二遍dfs，以a为起点，找到离a最远的点b，ab即为树中最远的两个点。证明：树的直径（最长路）的详细证明。要注意到一点：树中的任意两个点一定连通，且路径唯一。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
#include <cmath>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>

using namespace std;

#define FOR(i,k,n) for(int i=k;i<n;i++)
#define FORR(i,k,n) for(int i=k;i<=n;i++)
#define scan(a) scanf("%d",&a)
#define scann(a,b) scanf("%d%d",&a,&b)
#define scannn(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define mst(a,n)  memset(a,n,sizeof(a))
#define ll long long
#define N 10005
#define mod 1000000007
#define INF 0x3f3f3f3f

const double eps=1e-8;
const double pi=acos(-1.0);
int n,ans,start;
int vis[N];
vector<pair<int,int> > E[N];

void Dfs(int u,int len)
{
    int flag=0;
    for(int i=0;i<E[u].size();i++)
    {
        int v=E[u][i].first;
        int d=E[u][i].second;
        if(!vis[v])
        {
            flag=1;
            vis[v]=1;
            Dfs(v,len+d);
            vis[v]=0;
        }
    }
    if(!flag)
    {
        if(ans<len) ans=len,start=u;
    }
}

int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);

    scan(n);
    FOR(i,0,n-1)
    {
        int a,b,c;
        scannn(a,b,c);
        E[a].push_back(make_pair(b,c));
        E[b].push_back(make_pair(a,c));
    }
    ans=0;
    mst(vis,0);
    vis[1]=1;
    Dfs(1,0);

    ans=0;
    mst(vis,0);
    vis[start]=1;
    Dfs(start,0);

    //printf("%d\n",ans);
    printf("%d\n",ans*10+ans*(ans+1)/2);
    return 0;
}