제출 #553905

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
553905		elazarkoren	꿈 (IOI13_dreaming)	C++17	14 / 100	51 ms	65536 KiB

dreaming

#include "dreaming.h"
#include <bits/stdc++.h>
#define x first
#define y second
#define all(v) v.begin(), v.end()
#define chkmin(a, b) a = min(a, b)
#define chkmax(a, b) a = max(a, b)
using namespace std;
typedef long long ll;
typedef vector<ll> vi;
typedef vector<vi> vvi;
typedef pair<int, int> pii;
typedef vector<pii> vii;

const int MAX_N = 1e5 + 5;

vii tree[MAX_N];
int deg[MAX_N];

bitset<MAX_N> visited;

pii DfsFar(int node, int parent, int d) {
    visited[node] = true;
    pii p = {d, node};
    for (auto [neighbor, w] : tree[node]) {
        if (neighbor != parent) {
            chkmax(p, DfsFar(neighbor, node, d + w));
        }
    }
    return p;
}

bool DfsPath(int node, int parent, int end, vii &path) {
    if (node == end) {
        path.push_back({node, 0});
        return true;
    }
    for (auto [neighbor, w] : tree[node]) {
        if (neighbor != parent && DfsPath(neighbor, node, end, path)) {
            path.push_back({node, w});
            return true;
        }
    }
}

int ans = 0;

int FindRadius(int node) {
    int v = DfsFar(node, -1, 0).y;
    auto [diameter, u] = DfsFar(v, -1, 0);
    int x = diameter;
    int dist = 0;
    vii path;
    DfsPath(v, -1, u, path);
    for (auto [node, w] : path) {
        chkmin(x, max(dist, diameter - dist));
        dist += w;
    }
    chkmax(ans, diameter);
    return x;
}

int travelTime(int n, int m, int l, int a[], int b[], int t[]) {
    for (int i = 0; i < m; i++) {
        deg[a[i]]++, deg[b[i]]++;
    }
    for (int i = 0; i < n; i++) {
        tree[i].reserve(deg[i]);
    }
    for (int i = 0; i < m; i++) {
        tree[a[i]].push_back({b[i], t[i]});
        tree[b[i]].push_back({a[i], t[i]});
    }
    int s = 0;
    for (int i = 0; i < n; i++) {
        if (!visited[i]) {
            s += FindRadius(i) + l;
        }
    }
    chkmax(ans, s - l);
    return ans;
}

컴파일 시 표준 에러 (stderr) 메시지

dreaming.cpp: In function 'bool DfsPath(int, int, int, vii&)':
dreaming.cpp:44:1: warning: control reaches end of non-void function [-Wreturn-type]
   44 | }
      | ^

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