#include <bits/stdc++.h>
#include "dreaming.h"
#define f first
#define s second
using namespace std;
const int nax=1e5+5;
struct node{
int a,b,p;
};
vector<pair<int,int> > grafo[nax];
node dp[nax];
bitset<nax> v;
int mini=INT_MAX,pos;
int diam=0;
node max3(node a, int b, int pos){
if(b>=a.a)return {b,a.a,pos};
if(b>a.b) return {a.a,b,a.p};
return a;
}
int dfsb(int n, int a){
v[n]=1;
for(auto x:grafo[n]){
if(x.f==a)continue;
dp[n]=max3(dp[n],dfsb(x.f,n)+x.s,x.f);
}
diam=max(diam,dp[n].a+dp[n].b);
return dp[n].a;
}
void dfst(int n, int a, int d){
if(max(d,dp[n].a)<mini){
mini=max(d,dp[n].a);
pos=n;
}
for(auto x:grafo[n]){
if(x.f==a)continue;
int dd=max(d,(x.f!=dp[n].p?dp[n].a:dp[n].b))+x.s;
dfst(x.f,n,dd);
}
}
int travelTime(int N, int M, int L, int A[], int B[], int T[]){
for(int i=0;i<N;i++)dp[i]={0,0,-1};
/*
for(int i=0;i<M;i++){
grafo[A[i]-1].push_back({B[i]-1,T[i]});
grafo[B[i]-1].push_back({A[i]-1,T[i]});
}*/
vector<int> maxes;
for(int i=0;i<N;i++){
if(!v[i]){
dfsb(i,-1);
dfst(i,-1,0);
maxes.push_back(mini);
}
}
sort(maxes.begin(),maxes.end(),greater<int>());
int ans=diam;
if(maxes.size()>1)ans=max(ans,maxes[0]+maxes[1]+L);
if(maxes.size()>2)ans=max(ans,maxes[1]+maxes[2]+2*L);
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
38 ms |
5492 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
38 ms |
5492 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
38 ms |
5492 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
19 ms |
4860 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
38 ms |
5492 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
38 ms |
5492 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |