首先,要存储图,由于该图是稀疏图,所以用邻接表存储图
题目要求找出两个点,使其路程的花费最多,对于每个点,我们只需找到离这个点最远和次远的两个点,他们之间的路径就是最远路径,直接用dfs暴搜,如下:
int dfs(int u){
st[u]=true;
int d1=0,d2=0;//分别存储最远和次远路径
for(int i=h[u];i!=-1;i=ne[i]){
int j=e[i];
if(!st[j]){//如果该点没有被搜过
int s=dfs(j)+w[i];
if(s>=d1)d2=d1,d1=s;
else if(s>d2)d2=s;
}
}
ans=max(ans,d1+d2);//更新答案
return d1;//返回最长路径
}
搜的时候,以任意一个点为起点都行,我这里写的dfs(1),从一号点开始搜
在找到最远路径ans之后,路程的花费就是ans*10+(1+2+...+ans)=ans*10+ans*(1+ans)/2;
由于答案可能会很大,就输出ans*10+(long long)ans*(ans+1)/2即可
C++代码如下:
#include<iostream>
#include<cstring>
using namespace std;
const int N=100010,M=N*2;
typedef long long LL;
int h[N],e[M],w[M],ne[M],idx;
bool st[N];
int n,ans;
void add(int a,int b,int c){//加一条a->b的边,长度为c
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
int dfs(int u){
st[u]=true;//标记当前点已经被搜过
int d1=0,d2=0;//分别存储最大和次大距离
for(int i=h[u];i!=-1;i=ne[i]){
int j=e[i];
if(!st[j]){//如果当前点没有被搜过
int s=dfs(j)+w[i];
if(s>=d1)d2=d1,d1=s;
else if(s>d2)d2=s;
}
}
ans=max(ans,d1+d2);//更新ans
return d1;
}
int main(){
cin>>n;
memset(h,-1,sizeof h);//初始化
for(int i=0;i<n-1;i++){
int a,b,c;
cin>>a>>b>>c;
add(a,b,c),add(b,a,c);//双向边
}
dfs(1);
cout<<ans*10+(LL)ans*(ans+1)/2<<endl;
return 0;
}