[C++] BOJ (백준) 1074 : Z

문제

1074번: Z (acmicpc.net)

1074번: Z

코드

#include <iostream>

using namespace std;

int n, r, c;

void input() {
  ios_base::sync_with_stdio(false);
  cin.tie(nullptr);
  cin >> n >> r >> c;
}

void solve(int y, int x, int size, int cnt) {
  if (size == 0) {
    cout << cnt;
    return;
  }

  size--;
  int d = 1 << size;
  if (r >= y + d) {
    y += d;
    cnt += d * d * 2;
  }
  if (c >= x + d) {
    x += d;
    cnt += d * d;
  }
  solve(y, x, size, cnt);
}

int main() {
  input();
  solve(0, 0, n, 0);
  return 0;
}

설명

주어진 배열을 4등분하여 그 4개의 위치 중 (r,c)가 속한 위치를 찾는 재귀 함수를 만든다.