oj.uz

답안 #940804

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
940804 2024-03-07T16:23:59 Z duckindog 꿈 (IOI13_dreaming) C++17
0 / 100
 43 ms 13648 KB 
#include <bits/stdc++.h>
 
using namespace std;
 
#ifndef LOCAL
#include "dreaming.h"
#endif
 
const int N = 100'000 + 10;
vector<pair<int, int>> ad[N];
 
bool mk[N];
int dpin[N], dpout[N];
void dfs1(int u, int p = 0) { 
  mk[u] = true;
  for (const auto& [v, w] : ad[u]) { 
    if (v == p) continue;
    dfs1(v, u);
    dpin[u] = max(dpin[u], dpin[v] + w);
  }
}
int dfs2(int u, int p = 0) { 
  array<pair<int, int>, 2> best;
  best[0] = best[1] = {-1e9, -1};
  
  auto add = [&](pair<int, int> x) { 
    if (best[1] < x) swap(best[1], x);
    if (best[0] < best[1]) swap(best[0], best[1]);
  };
 
  add({dpout[u], u});
  for (const auto& [v, w] : ad[u]) if (v != p) add({dpin[v] + w, v});
 
  int ret = u;
  for (const auto& [v, w] : ad[u]) { 
    if (v == p) continue;
    dpout[v] = (best[0].second == v ? best[1].first : best[0].first) + w;
    int nv = dfs2(v, u);
    if (max(dpin[nv], dpout[nv]) < max(dpin[ret], dpout[ret])) ret = nv;
  }
  return ret;
}
 
int travelTime(int n, int m, int l, int A[],int B[],int T[]) { 
  for (int i = 1; i <= m; ++i) { 
    int u = A[i], v = B[i], w = T[i];
    ad[u].push_back({v, w});
    ad[v].push_back({u, w});
  }
  
  vector<int> ed;
  for (int i = 0; i < n; ++i) { 
    if (mk[i]) continue;
    dfs1(i);
    int x = dfs2(i);
    ed.push_back(x);
  }
  sort(ed.begin(), ed.end(), [&](const auto& a, const auto & b) {
    return max(dpin[a], dpout[a]) > max(dpin[b], dpout[b]);
  });
  
  for (const auto& x : ed) { 
    if (x == ed[0]) continue;
    ad[ed[0]].push_back({x, l});
    ad[x].push_back({ed[0], l});
  }
  memset(dpin, 0, sizeof dpin);
  memset(dpout, 0, sizeof dpout);
  dfs1(1); dfs2(1);
  int answer = 0;
  for (int i = 0; i < n; ++i) answer = max({answer, dpin[i], dpout[i]});
  
  return answer;
}
 
#ifdef LOCAL
int n, m, l;
int A[N], B[N], T[N];
int32_t main() { 
  cin.tie(0)->sync_with_stdio(0);
 
  cin >> n >> m >> l;
  for (int i = 1; i <= m; ++i) cin >> A[i] >> B[i] >> T[i];
  
  cout << travelTime(n, m, l, A, B, T) << "\n";
}
#endif

Subtask #1
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 43 ms 13648 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 1 ms 4444 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 23.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 43 ms 13648 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 18.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 34 ms 8396 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #5
0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 1 ms 4444 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #6
0 / 23.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 43 ms 13648 KB Output isn't correct
2 Halted 0 ms 0 KB -