东华oj进阶65 大臣的旅费(TLE)

问题描述:

很久以前，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最多花费的路费是多少。

比如：

输入

5
1 2 2
1 3 1
2 4 5
2 5 4

输出

135
即：大臣J从城市4到城市5要花费135的路费。

输入范例 :

5
1 2 2
1 3 1
2 4 5
2 5 4

输出范例 :

135

思路:

这类题归于求树的直径
求树的直径需要先对任一点进行DFS找到距离其最远的点，这个点必定是树的一个端点
然后再从这个最远点进行一次DFS，找到的必然是直径的另一个端点
最后根据直径计算出所需的消费
但是我这题不知道为什么又TLE了，可能是因为用的邻接矩阵存储的，但是传入的数据为稀疏图

代码实现:

#include<bits/stdc++.h>
using namespace std;

int n;
int **matrix;
int res=0;

bool *visited;

int maxFarNode;
int maxFarLen;

int GetCost(int n)
{
    return n*(n+1)/2+n*10;
}

//从node触发寻找最远的点
void DFS(int currLen,int node)
{
    if(visited[node]) return;
    visited[node]=true;
    for(int i=1;i<=n;i++)
    {
        if(visited[i]) continue;
        if(matrix[node][i]==-1) continue;

        int len=matrix[node][i];
        if(currLen+len>maxFarLen)
        {
            maxFarNode=i;
            maxFarLen=currLen+len;
        }

        DFS(currLen+len,i);
    }
}

int main()
{
    cin>>n;
    matrix=new int*[n+1];
    visited=new bool[n+1];

    for(int i=1;i<=n;i++)
    {
        matrix[i]=new int[n+1];
        visited[i]=false;
        for(int j=0;j<=n;j++)
        {
            matrix[i][j]=-1;
        }
    }

    int P,Q,D;
    for(int i=1;i<=n-1;i++)
    {
        cin>>P>>Q>>D;
        matrix[P][Q]=D;
        matrix[Q][P]=D;
    }

//    for(int i=1;i<=n;i++)
//    {
//        for(int j=1;j<=n;j++)
//            cout<<matrix[i][j]<<" ";
//        cout<<endl;
//    }

    DFS(0,1);
    fill(visited+1,visited+1+n,false);
    DFS(0,maxFarNode);


    cout<<GetCost(maxFarLen)<<endl;
}