제출 #286287

#제출 시각아이디문제언어결과실행 시간메모리
286287dolphingarlicHighway Tolls (IOI18_highway)C++14
51 / 100
232 ms262148 KiB
#include "highway.h"

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

vector<int> W;
vector<pair<int, int>> graph[90000], ord;

void dfs(int node, int parent = -1) {
    for (pair<int, int> i : graph[node]) if (i.first != parent) {
        ord.push_back(i);
        dfs(i.first, node);
    }
}

void find_pair(int N, vector<int> U, vector<int> V, int A, int B) {
    int M = int(U.size());
    if (M != N - 1) answer(0, 0);

    for (int i = 0; i < N; i++) graph[i].clear();
    ord.clear();
    for (int i = 0; i < M; i++) {
        graph[U[i]].push_back({V[i], i});
        graph[V[i]].push_back({U[i], i});
    }
    ord.push_back({0, -1});
    dfs(0);

    W.resize(M);
    for (int i = 0; i < M; i++) W[i] = 0;
    long long path_len = ask(W) / A;

    int l = 0, r = M;
    while (l != r) {
        int mid = (l + r) / 2;
        for (int i = 1; i <= mid; i++) W[ord[i].second] = 1;
        for (int i = mid + 1; i < M; i++) W[ord[i].second] = 0;
        long long dist = ask(W);
        if (dist == path_len * B) r = mid;
        else l = mid + 1;
    }
    int T = ord[l].first;

    ord.clear();
    ord.push_back({0, -1});
    dfs(T);

    l = 0, r = M;
    while (l != r) {
        int mid = (l + r) / 2;
        for (int i = 1; i <= mid; i++) W[ord[i].second] = 1;
        for (int i = mid + 1; i < M; i++) W[ord[i].second] = 0;
        long long dist = ask(W);
        if (dist == path_len * B) r = mid;
        else l = mid + 1;
    }
    int S = ord[l].first;

    answer(S, T);
}
#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...