# |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
255425 |
2020-08-01T00:41:23 Z |
babo |
Dreaming (IOI13_dreaming) |
C++14 |
|
0 ms |
0 KB |
#include <bits/stdc++.h>
#define L long long
using namespace std;
vector<L>v[100010],t[100010];
L n,len;
L diam;
multiset<L>rad;
L chk[100010];
L ord[100010],ordend[100010],ordtop;
L max2(L a,L b,L c){
if(a<=b&&a<=c) return b+c;
if(b<=c&&b<=a) return c+a;
if(c<=a&&c<=b) return a+b;
}
struct S{
L ma1,ma1d,ma2,ma2d;
S(){
ma1=ma1d=ma2=ma2d=0;
}
void insert(L dist,L loc){
if(dist>=ma1)
{
ma2=ma1;
ma2d=ma1d;
ma1=dist;
ma1d=loc;
}
else if(dist>=ma2)
{
ma2=dist;
ma2d=loc;
}
}
}S[100010];
void dfs(L x,L bu){
ordtop++;
ord[x]=ordtop;
L i;
S[x].insert(0,x);
for(i=0;i<v[x].size();i++)
{
if(v[x][i]!=bu)
{
chk[v[x][i]]=1;
dfs(v[x][i],x);
S[x].insert(S[v[x][i]].ma1+t[x][i],S[v[x][i]].ma1d);
}
}
diam=max(diam,S[x].ma1+S[x].ma2);
ordend[x]=ordtop;
}
L dfs2(L x,L bu,L give){
L i,ret=max2(give,S[x].ma1,S[x].ma2);
for(i=0;i<v[x].size();i++)
{
if(v[x][i]!=bu)
{
if(ord[v[x][i]]<=ord[S[x].ma1d]&&ord[S[x].ma1d]<=ordend[v[x][i]])
{
if(S[x].ma2d)
ret=min(ret,dfs2(v[x][i],x,max(give+t[x][i],S[x].ma2)));
else ret=min(ret,dfs2(v[x][i],x,give+t[x][i]));
}
else
ret=min(ret,dfs2(v[x][i],x,max(give+t[x][i],S[x].ma1)));
}
}
return ret;
}
void tree(L x){
chk[x]=1;
dfs(x,-1);
rad.insert(dfs2(x,-1,0));
}
int travelTime(int N,int M,int l,int A[],int B[],int T[]){
L i;
n=N,len=l;
for(i=0;i<M;i++)
{
v[A[i]].push_back(B[i]);
v[B[i]].push_back(A[i]);
t[A[i]].push_back(T[i]);
t[B[i]].push_back(T[i]);
}
for(i=0;i<N;i++)
{
if(!chk[i])
tree(i);
}
if(rad.size()==1) return (int)diam;
else return max((int)diam,(int)(len+*prev(rad.end())+*prev(prev((rad.end())))));
}
Compilation message
dreaming.cpp: In function 'void dfs(long long int, long long int)':
dreaming.cpp:46:11: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(i=0;i<v[x].size();i++)
~^~~~~~~~~~~~
dreaming.cpp: In function 'long long int dfs2(long long int, long long int, long long int)':
dreaming.cpp:61:11: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(i=0;i<v[x].size();i++)
~^~~~~~~~~~~~
dreaming.cpp: In function 'long long int max2(long long int, long long int, long long int)':
dreaming.cpp:18:1: warning: control reaches end of non-void function [-Wreturn-type]
}
^
/tmp/cc6EMcvB.o: In function `main':
grader.c:(.text.startup+0xa2): undefined reference to `travelTime'
collect2: error: ld returned 1 exit status