Submission #849297

#TimeUsernameProblemLanguageResultExecution timeMemory
849297abcvuitunggioDreaming (IOI13_dreaming)C++17
47 / 100
57 ms19024 KiB
#include "dreaming.h"
#include <bits/stdc++.h>
using namespace std;
vector <pair <int, int>> ke[100001];
int dp[100001],dp2[100001],ch[100001],mx;
priority_queue <int, vector <int>, greater <int>> q;
void dfs(int u, int p){
    ch[u]=1;
    for (auto [v,w]:ke[u])
        if (v!=p){
            dfs(v,u);
            dp[u]=max(dp[u],dp[v]+w);
        }
}
void dfs2(int u, int p){
    ch[u]=0;
    int mx=0,idx=-1,mx2=0;
    for (auto [v,w]:ke[u]){
        if (v==p)
            continue;
        if (dp[v]+w>mx){
            mx2=mx;
            mx=dp[v]+w;
            idx=v;
        }
        else
            mx2=max(mx2,dp[v]+w);
    }
    for (auto [v,w]:ke[u]){
        if (v==p)
            continue;
        dp2[v]=max(dp2[u],(v==idx?mx2:mx))+w;
        dfs2(v,u);
    }
}
int dfs3(int u, int p){
    ch[u]=1;
    int mn=dp[u];
    for (auto [v,w]:ke[u])
        if (v!=p)
            mn=min(mn,dfs3(v,u));
    return mn;
}
int travelTime(int N, int M, int L, int A[], int B[], int T[]) {
    for (int i=0;i<M;i++){
        ke[A[i]].emplace_back(B[i],T[i]);
        ke[B[i]].emplace_back(A[i],T[i]);
    }
    for (int i=0;i<N;i++)
        if (!ch[i])
            dfs(i,i);
    for (int i=0;i<N;i++)
        if (ch[i])
            dfs2(i,i);
    for (int i=0;i<N;i++){
        dp[i]=max(dp[i],dp2[i]);
        mx=max(mx,dp[i]);
    }
    for (int i=0;i<N;i++)
        if (!ch[i])
            q.push(dfs3(i,i));
    while (q.size()>1){
        int x=q.top();
        q.pop();
        int y=q.top();
        q.pop();
        mx=max(mx,x+y+L);
        q.push(min(max(x,y+L),max(x+L,y)));
    }
    return mx;
}
#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...