제출 #369111

#제출 시각아이디문제언어결과실행 시간메모리
369111KoD도시들 (IOI15_towns)C++17
35 / 100
26 ms512 KiB
#include <bits/stdc++.h>
#include "towns.h"

template <class T>
using Vec = std::vector<T>;

int memo[110][110];
int getDist(const int i, const int j) {
    if (i == j) return 0;
    if (memo[i][j] == 0) {
        memo[i][j] = memo[j][i] = getDistance(i, j);
    }
    return memo[i][j];
}

int hubDistance(int N, int sub) {
    for (int i = 0; i < N; ++i) {
        for (int j = 0; j < N; ++j) {
            memo[i][j] = 0;
        }
    }
    const auto farthest = [&](const int src) {
        Vec<int> v(N);
        std::iota(v.begin(), v.end(), 0);
        return *std::max_element(v.begin(), v.end(), [&](const int i, const int j) {
            return getDist(src, i) < getDist(src, j);
        });
    };
    const auto X = farthest(0);
    const auto Y = farthest(X);
    int R = 1000005;
    std::map<int, Vec<int>> map;
    Vec<int> near(N);
    const auto Diameter = getDist(X, Y);
    for (int u = 0; u < N; ++u) {
        const auto x = getDist(X, u);
        const auto z = ((x + getDist(0, u)) - getDist(X, 0)) / 2;
        const auto l = x - z;
        map[l].push_back(u);
        R = std::min(R, std::max(l, Diameter - l));
        near[u] = z;
    }
    Vec<Vec<int>> cand;
    {
        int left = 0;
        for (const auto &[l, v]: map) {
            if (std::max(left, N - (int) v.size() - left) * 2 <= N && std::max(l, Diameter - l) == R) {
                cand.push_back(v);
            }
            left += (int) v.size(); 
        }
    }
    if (cand.empty()) {
        return -R;
    }
    for (const auto &v: cand) {
        if ((int) v.size() * 2 <= N) {
            return R;
        }
    }
    const auto same = [&](const int x, const int y) {
        return getDist(x, y) < near[x] + near[y];
    };
    int comp, count = 0;
    for (const auto x: cand[0]) {
        if (count == 0) {
            comp = x;
        }
        else {
            if (same(x, comp)) {
                count += 1;
            }
            else {
                count -= 1;
            }
        }
    }
    int size = 0;
    for (const auto x: cand[0]) {
        if (same(comp, x)) {
            size += 1;
        }
    }
    return size * 2 <= N ? R : -R;
}

#ifdef LOCAL

int dist[110][110];

int getDistance(int i, int j) {
    return dist[i][j];
}

int main() {
    int sub, tests;
    std::cin >> sub >> tests;
    while (tests--) {
        int N;
        std::cin >> N;
        for (int i = 0; i < N; ++i) {
            for (int j = 0; j < N; ++j) {
                std::cin >> dist[i][j];
            }
        }
        std::cout << hubDistance(N, sub) << '\n';
    }
    return 0;
}

#endif

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

towns.cpp: In function 'int hubDistance(int, int)':
towns.cpp:16:28: warning: unused parameter 'sub' [-Wunused-parameter]
   16 | int hubDistance(int N, int sub) {
      |                        ~~~~^~~
#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...