Submission #376038

#	Time	Username	Problem	Language	Result	Execution time	Memory
376038		WLZ	꿈 (IOI13_dreaming)	C++14	18 / 100	69 ms	17516 KiB

dreaming

#include "dreaming.h"
#include <bits/stdc++.h>
using namespace std;

const int INF = 0x3f3f3f3f;

int cur_cc = 0;
vector< vector< pair<int, int> > > g;
vector<int> cc, d, mn, mx;

void dfs1(int u, int p) {
  cc[u] = cur_cc;
  for (auto& v : g[u]) {
    if (v.first != p) {
      d[v.first] = d[u] + v.second;
      dfs1(v.first, u);
    }
  }
  mx[u] = d[u];
  for (auto& v : g[u]) mx[u] = max(mx[u], mx[v.first]);
}

void dfs2(int u, int p, int len) {
  multiset<int> st;
  for (auto& v : g[u]) {
    if (v.first != p) {
      st.insert(mx[v.first]);
    }
  }
  for (auto& v : g[u]) {
    if (v.first != p) {
      st.erase(mx[v.first]);
      dfs2(v.first, u, max(len, (st.empty() ? 0 : *st.rbegin() - d[u])) + v.second);
      st.insert(mx[v.first]);
    }
  }
  mn[cc[u]] = min(mn[cc[u]], max(len, mx[u] - d[u]));
}

int travelTime(int N, int M, int L, int A[], int B[], int T[]) {
  g.resize(N);
  for (int i = 0; i < M; i++) g[A[i]].push_back({B[i], T[i]});
  for (int i = 0; i < M; i++) g[B[i]].push_back({A[i], T[i]});
  d.assign(N, 0);
  cc.assign(N, -1);
  mx.resize(N);
  for (int i = 0; i < N; i++) {
    if (cc[i] == -1) {
      dfs1(i, -1);
      cur_cc++;
    }
  }
  mn.assign(cur_cc, INF);
  for (int i = 0; i < N; i++) {
    if (mn[cc[i]] == INF) dfs2(i, -1, 0);
  }
  sort(mn.begin(), mn.end());
  if (cur_cc == 1) return mn[0];
  if (cur_cc == 2) return mn[0] + mn[1] + L;
  return max({*max_element(mn.begin(), mn.end()), mn[cur_cc - 2] + mn[cur_cc - 3] + 2 * L, mn[cur_cc - 1] + mn[cur_cc - 2] + L});
}

• Subtask 1: 0 / 14
• Subtask 2: 0 / 10
• Subtask 3: 0 / 23
• Subtask 4: 18 / 18
• Subtask 5: 0 / 12
• Subtask 6: 0 / 23