【历届试题】大臣的旅费

问题描述

很久以前，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的路费。

两次dfs，第一次找出最长路的一个端点，第二次便是从最长路的一个端点dfs出最长路；证明链接：http://www.cnblogs.com/wuyiqi/archive/2012/04/08/2437424.html

这个题目一直卡在过75%的数据，最后一组数据一直RE，郁闷了很久，网上也找了很多别人的博客，其中也有很多人是这样没过，还把代码贴出来了。有的人用Floyd算法，O(n^3)的复杂度，很显然不可能这么水！！！

后来看到有人说数据最后一组n为10000，总算是让我找到了错误的根源；题目没有详细的说明数据的范围，虽然报RE,很大的可能是是数组越界，但是想想两次dfs,如果数据量太大好像也不合理。。。orz...结果理想是美好的，现实是残酷的！！！哈哈。。。。还好过了

只要不用二维数组就好了，直接改成vector代替二维数组就好了。

代码：

#include<iostream>
#include<cstring>
#include<vector>
using namespace std;
struct node{
    int to,cost;
};
vector<node>G[<strong>10010</strong>];
int vis[10010];
int a,b,c,n;
int temp;
long long ans=-99999;
void dfs(int s,int dist)            //  深度优先搜索；
{
    if(ans<dist) ans=dist,temp=s;       //  保存最长路以及最长路端点；
    for(int i=0;i<G[s].size();i++){
        node p=G[s][i];
        if(!vis[p.to]&&p.cost){
            vis[p.to]=1;
            dist+=p.cost;
            dfs(p.to,dist);
            vis[p.to]=0;
            dist-=p.cost;
        }
    }
    return;
}
int main()
{
    cin>>n;
    node p;
    memset(vis,0,sizeof(vis));
    for(int i=0;i<n;i++) G[i].clear();
    for(int i=0;i<n-1;i++){
        cin>>a>>b>>c;
        p.to=b;
        p.cost=c;
        G[a].push_back(p);
        p.to=a;
        G[b].push_back(p);
    }
    vis[1]=1;
    dfs(1,0);               //  搜索最长路起始点；
    memset(vis,0,sizeof(vis));
    vis[temp]=1;
    ans=-99999;
    dfs(temp,0);            //  从起始点出发，搜索最长路；
    cout<<((11+10+ans)*ans)/2<<endl;
    return 0;
}

得75分代码：

#include<iostream>
#include<cstring>
using namespace std;
int Map[<strong>7010</strong>][<strong>7010</strong>];
int vis[<strong>7010</strong>];
int a,b,c,n;
int ans=-99999,temp;
void dfs(int s,int dist)            //  深度优先搜索；
{
    if(ans<dist) ans=dist,temp=s;       //  保存最长路以及最长路端点；
    for(int i=1;i<=n;i++){
        if(!vis[i]&&Map[s][i]){
            vis[i]=1;
            dist+=Map[s][i];
            dfs(i,dist);
            vis[i]=0;
            dist-=Map[s][i];
        }
    }
    return;
}
int main()
{
    cin>>n;
    memset(Map,0,sizeof(Map));
    memset(vis,0,sizeof(vis));
    for(int i=0;i<n-1;i++){
        cin>>a>>b>>c;
        Map[a][b]=Map[b][a]=c;
    }
    vis[1]=1;
    dfs(1,0);               //  搜索最长路起始点；
    memset(vis,0,sizeof(vis));
    vis[temp]=1;
    ans=-99999;
    dfs(temp,0);            //  从起始点出发，搜索最长路；
    cout<<((11+10+ans)*ans)/2<<endl;
    return 0;
}