#include"dreaming.h"
#include<iostream>
#include<vector>
#include<algorithm>
#include<unordered_map>
using namespace std;
typedef long long ll;
vector<pair<int,int>>nodes[100000];
bool vis[100000];
vector<int>node_list;
void dfs(int node){
if(vis[node])
return;
vis[node]=true;
node_list.push_back(node);
for(auto ne:nodes[node])
dfs(ne.first);
}
ll toll(int a,int b){
return ((ll)a<<32)+b;
}
unordered_map<ll,pair<int,int>>dp;
pair<int,int>get_max_dist(int node,int prev){
if(dp.count(toll(node,prev)))
return dp[toll(node,prev)];
pair<int,int>res={0,0};
for(auto ne:nodes[node]){
if(ne.first==prev)
continue;
{
int curr=get_max_dist(ne.first,node).first+ne.second;
if(curr>=res.first){
res.second=res.first;
res.first=curr;
}else if(curr>res.second){
res.second=curr;
}
}
}
dp[toll(node,prev)]=res;
return res;
}
pair<int,int> get_diameter(int root){
node_list.clear();
dfs(root);
/*cout<<"- ";
for(int i:node_list)
cout<<i<<" ";
cout<<"\n";*/
int res=1e9,diam=0;
for(int node:node_list){
//cout<<node<<": "<<get_max_dist(node,node)<<"\n";
auto r=get_max_dist(node,node);
res=min(res,r.first);
diam=max(diam,r.first+r.second);
}
return {res,diam};
}
int travelTime(int N,int M,int L,int A[],int B[],int T[]){
for(int i=0;i<M;i++){
nodes[A[i]].emplace_back(B[i],T[i]);
nodes[B[i]].emplace_back(A[i],T[i]);
}
vector<int>dists;
int diam=0;
for(int i=0;i<N;i++)
if(!vis[i]){
auto r=get_diameter(i);
dists.push_back(r.first);
diam=max(diam,r.second);
}
sort(dists.begin(),dists.end());
reverse(dists.begin(),dists.end());
if(dists.size()==1)
return max(dists[0],diam);
else if(dists.size()==2)
return max(dists[0]+dists[1]+L,diam);
else
return max(max(dists[0]+dists[1]+L,dists[1]+dists[2]+2*L),diam);
}
