# include "dreaming.h"
#include <iostream>
#include <vector>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;
const int Mxn=1e5+7;
vector<int> g[Mxn],cost[Mxn];
bool used[Mxn];
int di[Mxn];
void dfs(int v,int p){
for(int i=0;i<g[v].size();++i){
int to=g[v][i];
int c=cost[v][i];
if(to==p)continue;
dfs(to,v);
di[v]=max(di[v],di[to]+c);
}
}
int go(int v,int p,int cp){
int cs=-1;
int e=0;
for(int i=0;i<g[v].size();++i){
int to=g[v][i];
int c=cost[v][i];
if(to==p)continue;
if(cs==-1||di[cs]+e<di[to]+c){
cs=to;
e=c;
}
}
// cout<<v<<" "<<cp<<" "<<cs<<endl;
if(cs!=-1){
int q=0;
for(int i=0;i<g[v].size();++i){
int to=g[v][i];
int c=cost[v][i];
if(to==p||to==cs)continue;
q=max(q,c+di[to]);
}
if(max(di[cs]+e,max(q,cp))<=max(di[cs],max(q,cp)+e)){
return max(di[cs]+e,max(q,cp));
}
else{
return go(cs,v,max(q,cp)+e);
}
}
else{
return cp;
}
}
void dfs_fill(int v,int p){
used[v]=true;
for(int i=0;i<g[v].size();++i){
if(used[g[v][i]]==true)continue;
dfs_fill(g[v][i],v);
}
}
int travelTime (int N, int M, int L, int A[], int B[], int T[]) {
memset(di,0,sizeof(di));
memset(used,0,sizeof(used));
for(int i=0;i<M;++i){
g[A[i]].push_back(B[i]);
g[B[i]].push_back(A[i]);
cost[A[i]].push_back(T[i]);
cost[B[i]].push_back(T[i]);
}
vector<int> z;
for(int i=0;i<N;++i){
if(used[i]==false){
dfs(i,i);
z.push_back(go(i,i,0));
dfs_fill(i,i);
}
}
sort(z.begin(),z.end());
reverse(z.begin(),z.end());
if(z.size()==1)return z[0];
if(z.size()==2){
return z[0]+L+z[1];
}
return max(z[0]+L+z[1],L+L+z[1]+z[2]);
}
Compilation message
dreaming.cpp: In function 'void dfs(int, int)':
dreaming.cpp:14:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0;i<g[v].size();++i){
~^~~~~~~~~~~~
dreaming.cpp: In function 'int go(int, int, int)':
dreaming.cpp:25:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0;i<g[v].size();++i){
~^~~~~~~~~~~~
dreaming.cpp:37:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0;i<g[v].size();++i){
~^~~~~~~~~~~~
dreaming.cpp: In function 'void dfs_fill(int, int)':
dreaming.cpp:56:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0;i<g[v].size();++i){
~^~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
82 ms |
16376 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
82 ms |
16376 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
82 ms |
16376 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
42 ms |
10104 KB |
Output is correct |
2 |
Correct |
42 ms |
10232 KB |
Output is correct |
3 |
Correct |
53 ms |
10208 KB |
Output is correct |
4 |
Correct |
57 ms |
10240 KB |
Output is correct |
5 |
Correct |
47 ms |
10104 KB |
Output is correct |
6 |
Correct |
60 ms |
10868 KB |
Output is correct |
7 |
Correct |
48 ms |
10360 KB |
Output is correct |
8 |
Correct |
58 ms |
10104 KB |
Output is correct |
9 |
Correct |
47 ms |
10104 KB |
Output is correct |
10 |
Correct |
50 ms |
10368 KB |
Output is correct |
11 |
Correct |
7 ms |
5504 KB |
Output is correct |
12 |
Correct |
10 ms |
6268 KB |
Output is correct |
13 |
Correct |
10 ms |
6244 KB |
Output is correct |
14 |
Correct |
9 ms |
6140 KB |
Output is correct |
15 |
Correct |
9 ms |
6268 KB |
Output is correct |
16 |
Correct |
9 ms |
6136 KB |
Output is correct |
17 |
Correct |
9 ms |
6140 KB |
Output is correct |
18 |
Correct |
10 ms |
6272 KB |
Output is correct |
19 |
Correct |
9 ms |
6140 KB |
Output is correct |
20 |
Correct |
6 ms |
5504 KB |
Output is correct |
21 |
Correct |
6 ms |
5504 KB |
Output is correct |
22 |
Correct |
6 ms |
5504 KB |
Output is correct |
23 |
Correct |
10 ms |
6268 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
82 ms |
16376 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
82 ms |
16376 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |