코딩 문제 풀이 11(BackJoon) - Python

[kimTH65]

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

[kimTH65]

풀이

import math
N,r,c=map(int,input().split())
leng = int(math.pow(2,N))
r+=1
c+=1
stack = 0
while(1) :
    if(r>leng/2 and c>leng/2) : #4분면
        r-=leng/2 
        c-=leng/2
        stack += int(math.pow(leng,2))/4*3

    elif(r>leng/2 and c<=leng/2) : #3분면
        r-=leng/2 
        stack += int(math.pow(leng,2))/4*2

    elif(r<=leng/2 and c>leng/2) :#1분면
        c-=leng/2
        stack += int(math.pow(leng,2))/4*1

                                    #2분면 

    leng = leng/2
    if(leng <=1) :
        break
print(int(stack))