1074번 Z
https://www.acmicpc.net/problem/1074
문제
한수는 크기가 2^N × 2^N인 2차원 배열을 Z모양으로 탐색하려고 한다. 예를 들어, 2×2배열을 왼쪽 위칸, 오른쪽 위칸, 왼쪽 아래칸, 오른쪽 아래칸 순서대로 방문하면 Z모양이다.
N > 1인 경우, 배열을 크기가 2^(N-1) × 2^(N-1)로 4등분 한 후에 재귀적으로 순서대로 방문한다.
다음 예는 2^2 × 2^2 크기의 배열을 방문한 순서이다.
N이 주어졌을 때, r행 c열을 몇 번째로 방문하는지 출력하는 프로그램을 작성하시오.
다음은 N=3일 때의 예이다.
입력
첫째 줄에 정수 N, r, c가 주어진다.
출력
r행 c열을 몇 번째로 방문했는지 출력한다.
제한
1 ≤ N ≤ 15
0 ≤ r, c < 2N
예제 입력 1
2 3 1
예제 출력 1
11
예제 입력 2
3 7 7
예제 출력 2
63
예제 입력 3
1 0 0
예제 출력 3
0
예제 입력 4
4 7 7
예제 출력 4
63
예제 입력 5
10 511 511
예제 출력 5
262143
예제 입력 6
10 512 512
예제 출력 6
786432
풀이
-
어려운 재귀,,
-
바킹독님 강의 듣고 풀이를 정리해봤다
recursion 함수
- 2n X 2n에서 (r,c)를 방문하는 순서 반환하는 함수
- 파라미터 n, r, c
- base condition은 n == 0
- 1~4번 사각형 따로 처리해야 함
코드
# 1074 Z
import sys
input = sys.stdin.readline
def recursion(n, r, c): # 2^n * 2^n에서 (r,c)를 방문하는 순서를 반환하는 함수
# base condition
if n == 0: return 0
half = 2**n // 2 # 2**(n-1)
# 1번 사각형
if r < half and c < half:
return recursion(n-1, r, c)
# 2번 사각형
elif r < half and half <= c:
return half*half + recursion(n-1, r, c-half)
# 3번 사각형
elif r >= half and c < half:
return 2*half*half + recursion(n-1, r-half, c)
# 4번 사각형
elif r >= half and c >= half:
return 3*half*half + recursion(n-1, r-half, c-half)
N, R, C = map(int, input().split())
print(recursion(N, R, C))
