[Python] 1074번 Z

문제

한수는 크기가 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열을 몇 번째로 방문했는지 출력한다.

예제 입력

2 3 1

예제 출력

11

풀이

import sys

def solution(n, x, y):
    global result

    if not (x <= r < x + n) and not (y <= c < y + n):
        result += int(n ** 2)
        return

    if n == 2:
        if x == r and y == c:
            print(result)
            exit(0)
        result += 1

        if x == r and y + 1 == c:
            print(result)
            exit(0)
        result += 1

        if x + 1 == r and y == c:
            print(result)
            exit(0)
        result += 1

        if x + 1 == r and y + 1 == c:
            print(result)
            exit(0)
        result += 1

        print("result", result)
        return

    solution(n // 2, x, y)
    solution(n // 2, x, y + n // 2)
    solution(n // 2, x + n // 2, y)
    solution(n // 2, x + n // 2, y + n // 2)


if __name__ == '__main__':

    n, r, c = map(int, sys.stdin.readline().split())
    result = 0

    solution(2 ** n, 0, 0)