제출 #693309

#제출 시각아이디문제언어결과실행 시간메모리
693309PoonYaPat꿈 (IOI13_dreaming)C++14
18 / 100
37 ms11604 KiB
#include "dreaming.h"
#include <bits/stdc++.h>
using namespace std;
typedef pair<int,int> pii;

int dis_min,mmax1[100001],mmax2[100001],mnode[100001];
bool vis[100001];
vector<pii> adj[100001];
vector<int> v;

void dfs1(int x, int par, int root) {
    vis[x]=true;
    for (auto s : adj[x]) {
        if (s.first==par) continue;
        dfs1(s.first,x,root);
        int dis=mmax1[s.first]+s.second;

        if (mmax1[x]<=dis) {
            mmax2[x]=mmax1[x];
            mmax1[x]=dis;
            mnode[x]=s.first;
        } else if (mmax2[x]<dis) mmax2[x]=dis;
    }
}

void dfs2(int x, int par, int dis) {
    dis_min=min(dis_min,max(mmax1[x],dis));
    for (auto s : adj[x]) {
        if (s.first==par) continue;
        if (s.first==mnode[x]) dfs2(s.first,x,max(dis,mmax2[x])+s.second);
        else dfs2(s.first,x,max(dis,mmax1[x])+s.second);
    }
}

int travelTime(int N, int M, int L, int A[], int B[], int T[]) {
    for (int i=0; i<M; ++i) {
        adj[A[i]].push_back(pii(B[i],T[i]));
        adj[B[i]].push_back(pii(A[i],T[i]));
    }
    for (int i=0; i<N; ++i) {
        if (!vis[i]) {
            dfs1(i,-1,i);
            dis_min=mmax1[i];

            for (auto s : adj[i]) {
                if (s.first==mnode[i]) dfs2(mnode[i],i,mmax2[i]+s.second);
            }
            v.push_back(dis_min);
        }
    }

    sort(v.begin(),v.end(),greater<int>());
    if (v.size()==1) return v[0];
    else if (v.size()==2) return v[0]+v[1]+L;
    return max(v[0]+v[1]+L,v[1]+v[2]+2*L);
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...