1074번: Z (BOJ C/C++)

1074번: Z

사용 언어: C++

 

문제

한수는 크기가 2N × 2N인 2차원 배열을 Z모양으로 탐색하려고 한다. 예를 들어, 2×2배열을 왼쪽 위칸, 오른쪽 위칸, 왼쪽 아래칸, 오른쪽 아래칸 순서대로 방문하면 Z모양이다.

N > 1인 경우, 배열을 크기가 2N-1 × 2N-1로 4등분 한 후에 재귀적으로 순서대로 방문한다.

다음 예는 22 × 22 크기의 배열을 방문한 순서이다.

N이 주어졌을 때, r행 c열을 몇 번째로 방문하는지 출력하는 프로그램을 작성하시오.

다음은 N=3일 때의 예이다.

입력

첫째 줄에 정수 N, r, c가 주어진다.

출력

r행 c열을 몇 번째로 방문했는지 출력한다.

 

풀이

#include <bits/stdc++.h>
using namespace std;

int func(int N, int r, int c){
    if(N==0) return 0;
    int half = 1<<(N-1);
    //사각형을 네 부분으로 잘라서 (r,c)의 위치에 따라 분류
    if(r<half && c<half) return func(N-1, r, c);
    if(r<half && c>=half) return half*half + func(N-1, r, c-half);
    if(r>=half && c<half) return half*half*2 + func(N-1, r-half, c);
    return half*half*3 + func(N-1, r-half, c-half);
}

int main(void){
    ios::sync_with_stdio(0);
    cin.tie(0);

    int N, r, c;
    cin >> N >> r >> c;
    cout << func(N,r,c) << '\n';

    return 0;
}