Submission #1307373

#TimeUsernameProblemLanguageResultExecution timeMemory
1307373KickingKunChameleon's Love (JOI20_chameleon)C++20
0 / 100
36 ms444 KiB
#include "chameleon.h"
#include <bits/stdc++.h>

using namespace std;

namespace {

int variable_example = 1;

}  // namespace

vector <int> adj[1001];
int c[1001];

void dfs(int u, int color) {
  c[u] = color;
  for (int v: adj[u]) if (c[v] == -1) {
    dfs(v, 1 - color);
  }
}

bool haveEdge(int x, int y) {
  return find(adj[x].begin(), adj[x].end(), y) != adj[x].end();
}

void Solve(int N) {
  for (int i = 1; i <= 2 * N; i++) {
    adj[i].clear(); c[i] = -1;
  }

  for (int i = 1; i <= N; i++) {
    int pre = N;
    while (pre < 2 * N) {
      int low = pre + 1, high = 2 * N, nxt = -1;
      while (low <= high) {
        int mid = (low + high) >> 1;
        vector <int> ask = {i};
        for (int x = pre + 1; x <= mid; x++)
          ask.emplace_back(x);

        if (Query(ask) < ask.size()) 
          high = (nxt = mid) - 1;
        else low = mid + 1;
      }

      if (nxt == -1) break;
      adj[i].emplace_back(nxt);
      pre = nxt;
    }
  }

  for (int i = N + 1; i <= 2 * N; i++) {
    int pre = N + 1;
    while (pre > 1) {
      int low = 1, high = pre - 1, nxt = -1;
      while (low <= high) {
        int mid = (low + high) >> 1;
        vector <int> ask = {i};
        for (int x = mid; x < pre; x++)
          ask.emplace_back(x);

        if (Query(ask) < ask.size())
          low = (nxt = mid) + 1;
        else high = mid - 1;
      }
      
      if (nxt == -1) break;
      adj[i].emplace_back(nxt);
      pre = nxt;
    }
  }

  for (int i = 1; i <= 2 * N; i++) 
    if (c[i] == -1) dfs(i, 0);

  for (int i = 1; i <= 2 * N; i++) {
    if (adj[i].size() == 3) {
      int x = adj[i][0], y = adj[i][1], z = adj[i][2];
      int crush;

      if (Query({i, x, y}) == 1) {
        crush = z;
      }
      else if (Query({i, x, z}) == 1) {
        crush = y;
      }
      else {
        crush = x;
      }

      adj[i].erase(find(adj[i].begin(), adj[i].end(), crush));
      // cerr << i << " like " << crush << endl;
    }
  }

  for (int i = 1; i <= 2 * N; i++) if (c[i] == 0) {
    if (adj[i].size() == 1) Answer(i, adj[i][0]);
    else {
      int x = adj[i][0], y = adj[i][1];
      if (haveEdge(x, i)) Answer(i, x);
      else Answer(i, y);
    }
  }
}
#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...