#652309 (C++17) No.1149 色塗りゲーム

提出ソース

結果

問題	No.1149 色塗りゲーム
ユーザー	keijak
提出日時	2021-04-30 06:27:27
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	142 ms / 2,000 ms
コード長	4,918 bytes
コンパイル時間	2,917 ms
コンパイル使用メモリ	227,300 KB
実行使用メモリ	25,476 KB
平均クエリ数	20.92
最終ジャッジ日時	2024-07-17 11:19:17
合計ジャッジ時間	9,505 ms
ジャッジサーバーID （参考情報）	judge1 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	55 ms 24,568 KB
testcase_01	AC	53 ms 24,952 KB
testcase_02	AC	52 ms 24,836 KB
testcase_03	AC	53 ms 24,580 KB
testcase_04	AC	53 ms 24,452 KB
testcase_05	AC	54 ms 24,964 KB
testcase_06	AC	52 ms 24,452 KB
testcase_07	AC	53 ms 25,220 KB
testcase_08	AC	52 ms 24,836 KB
testcase_09	AC	53 ms 24,580 KB
testcase_10	AC	54 ms 24,580 KB
testcase_11	AC	54 ms 25,220 KB
testcase_12	AC	55 ms 24,580 KB
testcase_13	AC	55 ms 25,220 KB
testcase_14	AC	55 ms 25,220 KB
testcase_15	AC	57 ms 24,452 KB
testcase_16	AC	57 ms 25,220 KB
testcase_17	AC	57 ms 24,452 KB
testcase_18	AC	57 ms 24,964 KB
testcase_19	AC	57 ms 24,580 KB
testcase_20	AC	57 ms 24,452 KB
testcase_21	AC	57 ms 24,452 KB
testcase_22	AC	57 ms 25,220 KB
testcase_23	AC	58 ms 24,452 KB
testcase_24	AC	58 ms 24,964 KB
testcase_25	AC	59 ms 24,580 KB
testcase_26	AC	58 ms 24,836 KB
testcase_27	AC	60 ms 24,452 KB
testcase_28	AC	58 ms 24,964 KB
testcase_29	AC	59 ms 24,836 KB
testcase_30	AC	109 ms 24,452 KB
testcase_31	AC	104 ms 25,476 KB
testcase_32	AC	110 ms 25,220 KB
testcase_33	AC	111 ms 24,824 KB
testcase_34	AC	115 ms 24,836 KB
testcase_35	AC	116 ms 24,836 KB
testcase_36	AC	120 ms 25,220 KB
testcase_37	AC	121 ms 25,220 KB
testcase_38	AC	116 ms 24,836 KB
testcase_39	AC	127 ms 24,452 KB
testcase_40	AC	124 ms 24,580 KB
testcase_41	AC	127 ms 24,580 KB
testcase_42	AC	134 ms 24,836 KB
testcase_43	AC	131 ms 25,220 KB
testcase_44	AC	139 ms 24,836 KB
testcase_45	AC	137 ms 24,836 KB
testcase_46	AC	139 ms 24,580 KB
testcase_47	AC	137 ms 24,452 KB
testcase_48	AC	140 ms 24,836 KB
testcase_49	AC	142 ms 25,220 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
#define REP_(i, a_, b_, a, b, ...) \
  for (int i = (a), END_##i = (b); i < END_##i; ++i)
#define REP(i, ...) REP_(i, __VA_ARGS__, __VA_ARGS__, 0, __VA_ARGS__)
#define ALL(x) std::begin(x), std::end(x)
using i64 = long long;
using u64 = unsigned long long;

template <typename T, typename U>
inline bool chmax(T &a, U b) {
  return a < b and ((a = std::move(b)), true);
}
template <typename T, typename U>
inline bool chmin(T &a, U b) {
  return a > b and ((a = std::move(b)), true);
}
template <typename T>
inline int ssize(const T &a) {
  return (int)std::size(a);
}

template <typename T>
std::istream &operator>>(std::istream &is, std::vector<T> &a) {
  for (auto &x : a) is >> x;
  return is;
}
template <typename Container>
std::ostream &print_seq(const Container &a, std::string_view sep = " ",
                        std::string_view ends = "\n",
                        std::ostream &os = std::cout) {
  auto b = std::begin(a), e = std::end(a);
  for (auto it = std::begin(a); it != e; ++it) {
    if (it != b) os << sep;
    os << *it;
  }
  return os << ends;
}
template <typename T, typename = void>
struct is_iterable : std::false_type {};
template <typename T>
struct is_iterable<T, std::void_t<decltype(std::begin(std::declval<T>())),
                                  decltype(std::end(std::declval<T>()))>>
    : std::true_type {};

template <typename T, typename = std::enable_if_t<
                          is_iterable<T>::value &&
                          !std::is_same<T, std::string_view>::value &&
                          !std::is_same<T, std::string>::value>>
std::ostream &operator<<(std::ostream &os, const T &a) {
  return print_seq(a, ", ", "", (os << "{")) << "}";
}
template <typename T, typename U>
std::ostream &operator<<(std::ostream &os, const std::pair<T, U> &a) {
  return os << "(" << a.first << ", " << a.second << ")";
}

#ifdef ENABLE_DEBUG
template <typename T>
void pdebug(const T &value) {
  std::cerr << value;
}
template <typename T, typename... Ts>
void pdebug(const T &value, const Ts &...args) {
  pdebug(value);
  std::cerr << ", ";
  pdebug(args...);
}
#define DEBUG(...)                                   \
  do {                                               \
    std::cerr << " \033[33m (L" << __LINE__ << ") "; \
    std::cerr << #__VA_ARGS__ << ":\033[0m ";        \
    pdebug(__VA_ARGS__);                             \
    std::cerr << std::endl;                          \
  } while (0)
#else
#define pdebug(...)
#define DEBUG(...)
#endif

using namespace std;

int main() {
  int n;
  cin >> n;
  vector<bool> board(n, false);

  auto scan = [&]() -> vector<int> {
    vector<int> stones;
    int cur = 0;
    REP(i, n) {
      if (board[i]) {
        if (cur > 0) {
          stones.push_back(cur);
          cur = 0;
        }
      } else {
        cur++;
      }
    }
    if (cur > 0) stones.push_back(cur);
    sort(ALL(stones));
    return stones;
  };

  map<vector<int>, int> memo;

  auto nimber = [&](auto &rec, vector<int> stones) -> int {
    if (stones.empty()) return 0;
    if (ssize(stones) > 1) {
      int res = 0;
      for (auto x : stones) {
        assert(x > 0);
        res ^= rec(rec, vector<int>(1, x));
      }
      return res;
    }
    const int x = stones[0];
    if (x == 1) return 1;
    if (x == 2) return 2;
    assert(x > 2);

    auto it = memo.find(stones);
    if (it != memo.end()) return it->second;
    set<int> s;
    s.insert(rec(rec, vector<int>(1, x - 1)));
    s.insert(rec(rec, vector<int>(1, x - 2)));
    {
      vector<int> v = {x - 2, 1};
      while (v[0] > 0 and v[0] >= v[1]) {
        s.insert(rec(rec, v));
        v[0]--, v[1]++;
      }
    }
    {
      vector<int> v = {x - 3, 1};
      while (v[0] > 0 and v[0] >= v[1]) {
        s.insert(rec(rec, v));
        v[0]--, v[1]++;
      }
    }
    // Return the MEX.
    for (int i = 0;; ++i) {
      if (not s.count(i)) {
        memo[stones] = i;
        return i;
      }
    }
    assert(false);
  };

  auto next_move = [&]() -> pair<int, int> {
    // Take one.
    REP(i, n) {
      if (board[i]) continue;
      board[i] = true;
      int val = nimber(nimber, scan());
      if (val == 0) {
        return {1, i + 1};
      }
      board[i] = false;
    }

    // Take two.
    REP(i, n - 1) {
      if (board[i] or board[i + 1]) continue;
      board[i] = board[i + 1] = true;
      int val = nimber(nimber, scan());
      if (val == 0) {
        return {2, i + 1};
      }
      board[i] = board[i + 1] = false;
    }

    assert(false);
  };

  auto query = [&]() -> void {
    int k, x;
    tie(k, x) = next_move();
    cout << k << ' ' << x << endl;
    int t;
    cin >> t;
    if (t == 0 or t == 1) {
      exit(0);
    }
    assert(t == 3);
    cin >> k >> x;
    REP(i, k) {
      int p = x - 1 + i;
      assert(not board[p]);
      board[p] = true;
    }
  };

  while (true) {
    query();
  }
}

yukicoder

結果

テストケース

ソースコード