[Python] Z - 백준 (분할정복)

해당 글은 백준 1074번 문제 'Z'을 다룬다.

문제

https://www.acmicpc.net/problem/1074

1074번: Z

 

설명

해당 문제는 2x2 크기의 배열이 될 때까지 계속해서 2^n x 2^n 의 큰 사분면으로 생각하여,

각 사분면에 해당하는 visit을 더해주는 과정을 반복하면 된다.

 

코드

import sys
input = sys.stdin.readline
n, r, c = map(int, input().split())
visit = 0
while n != 0:
    n -= 1
    size = 2**n
    if r < size and c < size:       # 1사분면
        continue
    elif r < size and c >= size:    # 2사분면
        visit += (size*size)
        c -= size
    elif r >= size and c < size:    # 3사분면
        visit += (size*size)*2
        r -= size
    else:                           # 4사분면
        visit += (size*size)*3
        r -= size
        c -= size
print(visit)