#include "bits/stdc++.h"
#include "dreaming.h"
using namespace std;
const int maxn=1e5+5;
int idx,jog,ans;
int dp[maxn][2];
bool vis[maxn];
vector<pair<int,int>>E[maxn];
void calc(int nd,int pr){
vis[nd] = 1;
int a = 0,b = 0;
for(auto i : E[nd]){
if(i.first == pr) continue;
calc(i.first,nd);
if(dp[i.first][0] + i.second > a) b=a,a = dp[i.first][0] + i.second;
else if(dp[i.first][0] + i.second > b) b=dp[i.first][0] + i.second;
}
dp[nd][0] = a;
dp[nd][1] = b;
ans = max(ans,a + b);
}
void dfs(int nd,int pr,int yok){
vis[nd] = 1;
if(max(yok,dp[nd][0]) < jog){
idx = nd;
jog = max(yok,dp[nd][0]);
}
for(auto i : E[nd]){
if(i.first == pr) continue;
if(dp[i.first][0] + i.second == dp[nd][0]) dfs(i.first,nd,max(yok,dp[nd][1])+i.second);
else dfs(i.first,nd,max(yok,dp[nd][0])+i.second);
}
}
int travelTime(int N, int M, int L, int A[], int B[], int T[]) {
for(int i = 1;i <= M;i++){
E[A[i]].push_back({B[i],T[i]});
E[B[i]].push_back({A[i],T[i]});
}
for(int i = 1;i <= N;i++) if(!vis[i]) calc(i,0);
memset(vis,0,sizeof(vis));
vector<pair<int,int>>v;
for(int i = 1;i <= N;i++){
if(vis[i]) continue;
idx = -1;
jog = 1e9;
dfs(i,0,0);
v.push_back({jog,idx});
}
sort(v.begin(),v.end());
reverse(v.begin(),v.end());
if(M != N-1) ans = max(ans,v[0].first + v[1].first + L);
return ans;
}
// int main(){
// int n,m,l; cin >> n >> m >> l;
// int a[m + 1],b[m+1],t[m+1];
// for(int i = 1;i <= m;i++) cin >> a[i] >> b[i] >> t[i];
// cout<<travelTime(n,m,l,a,b,t);
// }
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
60 ms |
16036 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
2 ms |
2764 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
60 ms |
16036 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
28 ms |
6868 KB |
Output is correct |
2 |
Incorrect |
26 ms |
6820 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
2 ms |
2764 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
60 ms |
16036 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |