# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
536257 | groshi | Dreaming (IOI13_dreaming) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include<iostream>
#include<vector>
#include "dreaming.h"
using namespace std;
bool odw[110000];
struct wi{
vector<int> Q;
int maxx1=0,maxx2=0,skad1=0,skad2=0;
int maxx=0;
int ojc=0;
int czym=0;
int odw2=0;
int odw3=0;
}*w;
int dfs(int x)
{
odw[x]=1;
int wynik=0;
for(int i=0;i<w[x].Q.size();i+=2)
{
int pom=w[x].Q[i];
if(odw[pom]==1)
continue;
w[pom].ojc=x;
w[pom].czym=w[x].Q[i+1];
int ile=dfs(pom);
if(ile+w[x].Q[i+1]>w[x].maxx1)
{
w[x].maxx2=w[x].maxx1;
w[x].skad2=w[x].skad1;
w[x].maxx1=ile+w[x].Q[i+1];
w[x].skad1=pom;
}
else if(ile+w[x].Q[i+1]>w[x].maxx2)
{
w[x].maxx2=ile+w[x].Q[i+1];
w[x].skad2=pom;
}
wynik=max(wynik,ile+w[x].Q[i+1]);
}
return wynik;
}
void dfs2(int x)
{
w[x].odw2=1;
for(int i=0;i<w[x].Q.size();i+=2)
{
int pom=w[x].Q[i];
if(w[pom].odw2==1)
continue;
if(w[x].skad1==pom && w[x].maxx2+w[x].Q[i+1]>w[pom].maxx1)
{
w[pom].maxx2=w[pom].maxx1;
w[pom].skad2=w[pom].skad1;
w[pom].maxx1=w[x].maxx2+w[x].Q[i+1];
w[pom].skad1=x;
}
else if(w[x].skad1==pom && w[x].maxx2+w[x].Q[i+1]>w[pom].maxx2)
{
w[pom].maxx2=w[x].maxx2+w[x].Q[i+1];
w[pom].skad2=x;
}
else if(w[x].skad1!=pom && w[x].maxx1+w[x].Q[i+1]>w[pom].maxx1)
{
w[pom].maxx2=w[pom].maxx1;
w[pom].skad2=w[pom].skad1;
w[pom].maxx1=w[x].maxx1+w[x].Q[i+1];
w[pom].skad1=x;
}
else if(w[x].skad1!=pom && w[x].maxx2+w[x].Q[i+1]>w[pom].maxx2)
{
w[pom].maxx2=w[x].maxx2+w[x].Q[i+1];
w[pom].skad2=x;
}
dfs2(pom);
}
}
int dfs3(int x)
{
w[x].odw3=1;
int wypisz=0;
wypisz=w[x].maxx1;
for(int i=0;i<w[x].Q.size();i+=2)
{
int pom=w[x].Q[i];
if(w[pom].odw3==1)
continue;
int cos=dfs3(pom);
wypisz=min(wypisz,cos);
}
return wypisz;
}
int travelTime(int n,int m,int l,int A[],int B[],int T[])
{
w=new wi[n+3];
for(int i=0;i<m;i++)
{
w[A[i]].Q.push_back(B[i]);
w[A[i]].Q.push_back(T[i]);
w[B[i]].Q.push_back(A[i]);
w[B[i]].Q.push_back(T[i]);
}
for(int i=0;i<n;i++)
{
if(odw[i]==1)
continue;
int maxx=dfs(i);
}
for(int i=0;i<n;i++)
{
if(w[i].odw2==1)
continue;
dfs2(i);
}
vector<int> drogi;
int jeden=0,dwa=0,trzy=0;
for(int i=0;i<n;i++)
{
if(w[i].odw3==1)
continue;
int ile=dfs3(i);
drogi.push_back(ile);
}
sort(drogi.begin(),drogi.end());
int wypisz=drogi.back();
if(drogi.size()>1)
wypisz=max(wypisz,drogi[drogi.size()-2]+drogi.back()+l);
if(drogi.size()>2)
wypisz=max(wypisz.drogi[drogi.size()-2]+drogi[drogi.size()-3]+2*l);
return wypisz;
return 0;
}