답안 #990562

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
990562 2024-05-30T16:13:11 Z jakobrs Ancient Machine 2 (JOI23_ancient2) C++17
0 / 100
27 ms 528 KB
#include "ancient2.h"

#include <bitset>
#include <memory>
#include <numeric>
#include <string>
#include <vector>

void gauss(std::bitset<1001> (&matrix)[1000]) {
  for (int col = 0; col < 1000; col++) {
    int pivot_row;
    for (int row = col; row < 1000; row++) {
      if (matrix[row][col]) {
        pivot_row = row;
        break;
      }
    }
    std::swap(matrix[col], matrix[pivot_row]);
    for (int row = col + 1; row < 1000; row++)
      if (matrix[row][col])
        matrix[row] ^= matrix[col];
  }
  for (int col = 999; col >= 0; col--) {
    for (int row = col - 1; row >= 0; row--)
      if (matrix[row][col])
        matrix[row] ^= matrix[col];
  }
}

std::string Solve(int N) {
  int m = 57 * 2;
  std::vector<int> a(m), b(m);
  std::vector<bool> results;
  int acc_phi = 0;

  auto matrix = std::make_unique<std::bitset<1001>[]>(1000);
  int cur = 0;
  for (int i = 1; i < 58; i++) {
    for (int j = 0; j < i; j++) {
      a[j] = j + 1;
      a[j + i] = j + i + 1;
      b[j] = j + 1;
      b[j + i] = j + i + 1;
    }
    a[i - 1] = 0;
    a[i - 1 + i] = i;
    b[i - 1] = 0;
    b[i - 1 + i] = i;

    int phi = 0;
    for (int j = 1; j <= i; j++)
      if (std::gcd(i, j) == 1)
        phi += 1;

    for (int j = 0; j < phi; j++) {
      for (int k = 0; k < 1000; k++) {
        if (k % i == j) {
          matrix[results.size()][k] = 1;
        }
      }

      std::swap(b[j], b[j + i]);
      auto res = Query(m, a, b);
      matrix[cur++] = res >= i;
      std::swap(b[j], b[j + i]);
    }
  }

  gauss(*(std::bitset<1001>(*)[1000])matrix.get());

  std::string s;

  for (int i = 0; i < 1000; i++)
    s.push_back(matrix[i][1000] ? '1' : '0');

  s.erase(s.begin() + N, s.end());

  return s;
}

Compilation message

ancient2.cpp: In function 'std::string Solve(int)':
ancient2.cpp:34:7: warning: unused variable 'acc_phi' [-Wunused-variable]
   34 |   int acc_phi = 0;
      |       ^~~~~~~
# 결과 실행 시간 메모리 Grader output
1 Incorrect 27 ms 528 KB Wrong Answer [3]
2 Halted 0 ms 0 KB -