1. 程式人生 > >藍橋杯 PREV-9 大臣的旅費 (兩次DFS)

藍橋杯 PREV-9 大臣的旅費 (兩次DFS)

題意:求樹中最遠的兩個點的距離。

解題方案:第一遍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;
}