| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 405660 | Alex_tz307 | 꿈 (IOI13_dreaming) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
const int MAXN = 1e5;
vector<pair<int,int>> G[MAXN];
int p[MAXN], dp1[MAXN], dp2[MAXN], node, val, dp[MAXN], best_node, best_dist;
bitset<MAXN> viz;
vector<int> nodes;
void max_self(int &a, int b) {
a = max(a, b);
}
void dfs1(int u) {
viz[u] = true;
for (auto v : G[u])
if (!viz[v.first]) {
p[v.first] = u;
dfs1(v.first);
if (dp1[u] <= dp1[v.first] + v.second) {
dp2[u] = dp1[u];
dp1[u] = dp1[v.first] + v.second;
} else max_self(dp2[u], dp1[v.first] + v.second);
}
}
void dfs2(int u, int dist) {
dp[u] = max(dp1[u], dist);
if (dp[u] < val) {
val = dp[u];
node = u;
}
for (auto v : G[u])
if (v.first != p[u]) {
if (dp1[u] == dp1[v.first] + v.second)
dfs2(v.first, max(dp2[u], dist) + v.second);
else dfs2(v.first, max(dp1[u], dist) + v.second);
}
}
void find_diameter(int u, int parent, int dist) {
if (best_dist < dist) {
best_dist = dist;
best_node = u;
}
for (auto v : G[u])
if (v.first != parent)
find_diameter(v.first, u, dist + v.second);
}
int travelTime(int N, int M, int L, int A[], int B[], int T[]) {
for (int i = 0; i < M; ++i) {
int u = A[i], v = B[i], w = T[i];
G[u].emplace_back(v, w);
G[v].emplace_back(u, w);
}
int max_val = -1, u = -1;
for (int i = 0; i < N; ++i)
if (!viz[i]) {
p[i] = -1;
dfs1(i);
node = -1;
val = INF;
dfs2(i, 0);
nodes.emplace_back(node);
if (max_val < dp[node]) {
max_val = dp[node];
u = node;
}
}
for (int v : nodes)
if (u != v) {
G[u].emplace_back(v, L);
G[v].emplace_back(u, L);
}
best_node = best_dist = -1;
find_diameter(0, -1, 0);
best_dist = -1;
find_diameter(best_node, -1, 0);
return best_dist;
}
void fastIO() {
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
}
int N, M, L, A[MAXN], B[MAXN], T[MAXN];
void solve() {
cin >> N >> M >> L;
for (int i = 0; i < M; ++i)
cin >> A[i] >> B[i] >> T[i];
cout << travelTime(N, M, L, A, B, T) << '\n';
}
int main() {
fastIO();
solve();
return 0;
}