제출 #217321

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
217321		fedoseevtimofey	카멜레온의 사랑 (JOI20_chameleon)	C++14	0 / 100	93 ms	5752 KiB

chameleon

#include "chameleon.h"
#include <iostream>
#include <string>
#include <vector>
#include <queue>
#include <deque>
#include <stack>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <random>
#include <iomanip>
#include <functional>
#include <cassert>

using namespace std;

typedef long long ll;

void Solve(int n) {
  vector <vector <int>> g(2 * n);
  
  int cnt_qr = 0;
  map <vector <int>, int> ans;
  function <int(vector <int>)> my_query = [&] (const vector <int> &p) {
    vector <int> a = p;
    sort(a.begin(), a.end());
    for (int &x : a) ++x;
    if (!ans.count(a)) {
      ++cnt_qr;
      ans[a] = Query(a);
    }
    return ans[a];
  };
  function <void(int, vector <int>)> rec = [&] (int v, vector <int> a) {
    if (a.size() == 1) {
      g[v].push_back(a[0]);
      g[a[0]].push_back(v);
    } else {
      int m = a.size() / 2;
      vector <int> b, c;
      for (int i = 0; i < m; ++i) b.push_back(a[i]);
      for (int i = m; i < (int)a.size(); ++i) c.push_back(a[i]);
      b.push_back(v);
      c.push_back(v);
      if (my_query(b) != (int)b.size()) {
        b.pop_back();
        rec(v, b);
      }
      if (my_query(c) != (int)c.size()) {
        c.pop_back();
        rec(v, c);
      }
    }
  };
  
  
  vector <int> a;
  for (int i = n; i < 2 * n; ++i) a.push_back(i);
  for (int i = 0; i < n; ++i) {
    rec(i, a);
  }

  for (int i = 0; i < 2 * n; ++i) {
    sort(g[i].begin(), g[i].end());
    g[i].resize(unique(g[i].begin(), g[i].end()) - g[i].begin());
  }
  vector <int> who(2 * n, -1);
  vector <set <int>> bad(2 * n);
  for (int i = 0; i < 2 * n; ++i) {
    if (g[i].size() == 1) {
      who[i] = g[i][0];
      who[g[i][0]] = i;
    } else if (g[i].size() == 3) {
      for (int j = 0; j < 3; ++j) {
        vector <int> p = {i};
        for (int k = 0; k < 3; ++k) {
          if (k != j) p.push_back(g[i][k]);
        }
        if (my_query(p) == 1) {
          bad[i].insert(g[i][j]);
          bad[g[i][j]].insert(i);
        }
      }
    } else {
      assert(false);
    }
  }
  for (int i = 0; i < 2 * n; ++i) {
    if (who[i] == -1) {
      for (int j = 0; j < (int)g[i].size(); ++j) {
        if (!bad[i].count(g[i][j])) {
          who[i] = g[i][j];
        }
      }
    }
  }

  for (int i = 0; i < 2 * n; ++i) {
    if (who[i] > i) {
      Answer(i + 1, who[i] + 1);
    }
  }
}

• Subtask 1: 0 / 4
• Subtask 2: 0 / 20
• Subtask 3: 0 / 20
• Subtask 4: 0 / 20
• Subtask 5: 0 / 36