oj.uz

Submission #221593

# Submission time Handle Problem Language Result Execution time Memory
221593 2020-04-10T14:06:05 Z rama_pang ICC (CEOI16_icc) C++14
100 / 100
 183 ms 640 KB 
#include "icc.h"
#include <bits/stdc++.h>
using namespace std;

mt19937 rnd(chrono::high_resolution_clock::now().time_since_epoch().count());
vector<vector<int>> components;

int Query(vector<int> A, vector<int> B) {
  if (A.empty() || B.empty()) return 0;
  return query(A.size(), B.size(), A.data(), B.data());
}

int QueryComponent(vector<int> A, vector<int> B) {
  vector<int> P, Q;
  for (auto i : A) {
    for (auto j : components[i]) {
      P.emplace_back(j);
    }
  }
  for (auto i : B) {
    for (auto j : components[i]) {
      Q.emplace_back(j);
    }
  }
  return Query(P, Q);
}

void Answer(int A, int B) {
  return setRoad(A, B);
}

int Split(int L, int R, vector<int> &A, vector<int> &B, int depth, bool Check) {
  if (depth == 0) {
    int mid = (L + R) / 2;
    for (int i = L; i <= mid; i++) A.emplace_back(i);
    for (int i = mid + 1; i <= R; i++) B.emplace_back(i);
    return max(mid - L + 1, R - mid);
  }
  int mid = (L + R) / 2;
  if (Check) {
    return max(Split(L, mid, A, B, depth - 1, Check), Split(mid + 1, R, A, B, depth - 1, Check));
  } else {
    vector<int> tA, tB;
    Split(L, mid, tA, tB, depth - 1, true);
    if (QueryComponent(tA, tB)) {
      return Split(L, mid, A, B, depth - 1, Check);
    } else {
      return Split(mid + 1, R, A, B, depth - 1, Check);
    }
  }
}

pair<int, int> BinarySearch(vector<int> A, vector<int> B, bool onComponent) {
  pair<int, int> res = {-1, -1};

  {
    int lo = 0, hi = A.size() - 1;
    while (lo < hi) {
      int mid = (lo + hi) / 2;
      vector<int> halfA;
      for (int i = 0; i <= mid; i++) {
        halfA.emplace_back(A[i]);
      }
      int res = (onComponent ? QueryComponent(halfA, B) : Query(halfA, B));
      if (res) {
        hi = mid;
      } else {
        lo = mid + 1;
      }
    }
    res.first = A[lo];
  }

  {
    int lo = 0, hi = B.size() - 1;
    while (lo < hi) {
      int mid = (lo + hi) / 2;
      vector<int> halfB;
      for (int i = 0; i <= mid; i++) {
        halfB.emplace_back(B[i]);
      }
      int res = (onComponent ? QueryComponent(halfB, A) : Query(halfB, A));
      if (res) {
        hi = mid;
      } else {
        lo = mid + 1;
      }
    }
    res.second = B[lo];
  }

  return res;
}

void MergeComponent(int C1, int C2) {
  vector<int> Merged;
  for (auto i : components[C1]) Merged.emplace_back(i);
  for (auto i : components[C2]) Merged.emplace_back(i);
  components[C1].clear();
  components[C2].clear();
  sort(begin(components), end(components), [&](const vector<int> &a, const vector<int> &b) {
    return a.size() > b.size();
  });

  while (components.back().empty()) components.pop_back();
  components.emplace_back(Merged);
}

void run(int N) {
  for (int i = 1; i <= N; i++) {
    components.emplace_back(vector<int>{i});
  }

  for (int i = 1; i < N; i++) {
    shuffle(begin(components), end(components), rnd);
    vector<int> A, B; // find two components that are connected by new edge
    for (int depth = 0; ; depth++) {
      int sz = Split(0, components.size() - 1, A, B, depth, true);
      if (sz == 1) break;
      int res = QueryComponent(A, B);
      A.clear(), B.clear();
      if (res) {
        Split(0, components.size() - 1, A, B, depth, false);
        break;
      }
    }

    // Binary Search to find edge's endpoint
    auto TwoComp = BinarySearch(A, B, true);
    auto Edge = BinarySearch(components[TwoComp.first], components[TwoComp.second], false);
    Answer(Edge.first, Edge.second);
    MergeComponent(TwoComp.first, TwoComp.second);
  }
}

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 12 ms 512 KB Ok! 111 queries used.
2 Correct 11 ms 512 KB Ok! 106 queries used.

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 49 ms 512 KB Ok! 605 queries used.
2 Correct 58 ms 512 KB Ok! 567 queries used.
3 Correct 49 ms 512 KB Ok! 596 queries used.

Subtask #3
22.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 155 ms 632 KB Ok! 1528 queries used.
2 Correct 165 ms 512 KB Ok! 1467 queries used.
3 Correct 165 ms 512 KB Ok! 1635 queries used.
4 Correct 156 ms 632 KB Ok! 1510 queries used.

Subtask #4
21.0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 168 ms 512 KB Ok! 1590 queries used.
2 Correct 171 ms 512 KB Ok! 1597 queries used.
3 Correct 183 ms 512 KB Ok! 1678 queries used.
4 Correct 167 ms 572 KB Ok! 1551 queries used.

Subtask #5
29.0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 168 ms 512 KB Ok! 1596 queries used.
2 Correct 169 ms 512 KB Ok! 1591 queries used.
3 Correct 170 ms 512 KB Ok! 1532 queries used.
4 Correct 181 ms 640 KB Ok! 1646 queries used.
5 Correct 152 ms 572 KB Ok! 1523 queries used.
6 Correct 145 ms 512 KB Ok! 1472 queries used.

Subtask #6
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 168 ms 632 KB Ok! 1570 queries used.
2 Correct 175 ms 512 KB Ok! 1573 queries used.