历届试题 大臣的旅费（树的直径）

时间限制：1.0s 内存限制：256.0MB
提交此题 锦囊1 锦囊2
问题描述
很久以前，T王国空前繁荣。为了更好地管理国家，王国修建了大量的快速路，用于连接首都和王国内的各大城市。

为节省经费，T国的大臣们经过思考，制定了一套优秀的修建方案，使得任何一个大城市都能从首都直接或者通过其他大城市间接到达。同时，如果不重复经过大城市，从首都到达每个大城市的方案都是唯一的。

J是T国重要大臣，他巡查于各大城市之间，体察民情。所以，从一个城市马不停蹄地到另一个城市成了J最常做的事情。他有一个钱袋，用于存放往来城市间的路费。

聪明的J发现，如果不在某个城市停下来修整，在连续行进过程中，他所花的路费与他已走过的距离有关，在走第x千米到第x+1千米这一千米中（x是整数），他花费的路费是x+10这么多。也就是说走1千米花费11，走2千米要花费23。

J大臣想知道：他从某一个城市出发，中间不休息，到达另一个城市，所有可能花费的路费中最多是多少呢？

输入格式
输入的第一行包含一个整数n，表示包括首都在内的T王国的城市数

城市从1开始依次编号，1号城市为首都。

接下来n-1行，描述T国的高速路（T国的高速路一定是n-1条）

每行三个整数Pi, Qi, Di，表示城市Pi和城市Qi之间有一条高速路，长度为Di千米。

输出格式
输出一个整数，表示大臣J最多花费的路费是多少。

样例输入1
5
1 2 2
1 3 1
2 4 5
2 5 4
样例输出1
135
输出格式
大臣J从城市4到城市5要花费135的路费。

分析：该题目题意就是要求树的直径。
求树直径原理：以任意点w开始，先做一次DFS（BFS），找到最远点v，然后以此点v，进行一次DFS（BFS），找到最远点u，u到v就是树的直径，记做d(u,v)。

1. dfs写法

#include <bits/stdc++.h>
using namespace std;
#define mem(a,n) memset(a,n,sizeof(a))
#define rep(i,a,n) for(int i=a;i<n;i++)
#define pb push_back
#define IO ios::sync_with_stdio(false)
#define fre freopen("in.txt","r",stdin)
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef long long ll;
const double PI=acos(-1.0);
const double eps=1e-8;
const int INF=0x3f3f3f3f;
const int MOD=1e9+7;
const int N=1e4+5;
const int dir[4][2]= {-1,0,1,0,0,-1,0,1};
struct node
{
    int v,w;
};
vector<node>e[N];
int n,ans_node,ans_d;
bool vis[N];
void dfs(int u,int d)
{
    if(d>ans_d)
    {
        ans_d=d;
        ans_node=u;
    }
    rep(i,0,e[u].size())
    {
        if(vis[e[u][i].v]) continue;
        vis[e[u][i].v]=1;
        dfs(e[u][i].v,d+e[u][i].w);
        vis[e[u][i].v]=0;
    }
}
int main()
{
    IO;
    //fre;
    cin>>n;
    rep(i,0,n-1)
    {
        int u,v,w;
        cin>>u>>v>>w;
        e[u].pb((node){v,w});///
        e[v].pb((node){u,w});///构造邻接表
    }
    ans_d=-1;
    mem(vis,0);
    vis[1]=1;
    dfs(1,0);///第一次dfs起点为1
    ans_d=-1;
    mem(vis,0);
    vis[ans_node]=1;
    dfs(ans_node,0);///第二次dfs起点为上一次找到的最远结点ans_node
    //cout<<"ans_d = "<<ans_d<<endl;
    ans_d=ans_d*11+(ans_d*(ans_d-1))/2;
    cout<<ans_d<<endl;
    return 0;
}

参考：http://www.itwendao.com/article/detail/205364.html

2.树状dp写法

#include <bits/stdc++.h>
using namespace std;
#define mem(a,n) memset(a,n,sizeof(a))
#define rep(i,a,n) for(int i=a;i<n;i++)
#define pb push_back
#define IO ios::sync_with_stdio(false)
#define fre freopen("in.txt","r",stdin)
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef long long ll;
const double PI=acos(-1.0);
const double eps=1e-8;
const int INF=0x3f3f3f3f;
const int MOD=1e9+7;
const int N=2e2+5;
const int dir[4][2]= {-1,0,1,0,0,-1,0,1};
int first[N], second[N];///first 存储到达最远结点的长度，second记录次长度
int res = 0;
struct node
{
    int v,w;
};
vector<node>g[N];
void dfs(int u, int father)
{
    rep(i,0,g[u].size())
    {
        int& v=g[u][i].v;
        if(v==father) continue;///子结点 == 父结点编号 边重复不需要重复扫
        dfs(v,u);
        /**当子结点的最长边+g[u][i].w（加上到父结点的距离）大于父结点的最长边 更新父结点的最长边和次长边**/
        if(first[v]+g[u][i].w>=first[u]) second[u]=first[u],first[u]=first[v]+g[u][i].w;
        else if(first[v]+g[u][i].w>second[u]) second[u]=first[v]+g[u][i].w;
    }
    /**在以当前结点为根节点的子树中，最远两个结点的距离为first＋second**/
    res = max(res, first[u]+second[u]);
    return ;
}
int main()
{
    //fre;
    int n,u,v,w;
    scanf("%d",&n);
    rep(i,0,n)
    {
        scanf("%d %d %d",&u,&v,&w);
        g[u].pb((node){v,w});
        g[v].pb((node){u,w});
    }
    dfs(1,0);
    res=res*11+(res*(res-1))/2;
    printf("%d\n",res);
}