other/magic_square.hpp

• View this file on GitHub
• Last update: 2022-11-27 18:59:55+09:00
• Include: #include "other/magic_square.hpp"

Verified with

• test/mytest/magic_square.test.cpp

Code

// とりあえず奇数のみ対応する
vvc<int> magic_square(int N) {
  assert(N % 2 == 1);
  vv(int, A, N, N);
  int x = 0, y = N / 2;
  FOR(i, N * N) {
    A[x][y] = i + 1;
    int nx = (x == 0 ? N - 1 : x - 1);
    int ny = (y == N - 1 ? 0 : y + 1);
    if (A[nx][ny] != 0) { nx = (x == N - 1 ? 0 : x + 1), ny = y; }
    tie(x, y) = mp(nx, ny);
  }
  return A;
}

#line 1 "other/magic_square.hpp"
// とりあえず奇数のみ対応する
vvc<int> magic_square(int N) {
  assert(N % 2 == 1);
  vv(int, A, N, N);
  int x = 0, y = N / 2;
  FOR(i, N * N) {
    A[x][y] = i + 1;
    int nx = (x == 0 ? N - 1 : x - 1);
    int ny = (y == N - 1 ? 0 : y + 1);
    if (A[nx][ny] != 0) { nx = (x == N - 1 ? 0 : x + 1), ny = y; }
    tie(x, y) = mp(nx, ny);
  }
  return A;
}

Back to top page