[백준] 1074 Z python

www.acmicpc.net/problem/1074

1074번: Z

N, r, c = map(int, input().split())
def z(N, r, c):
    result = 0
    while N > 0:
        if N == 1:
            result += r * 2 + c
            break
        else:
            flag = 4 ** (N - 1)
            half = (2 ** N) >> 1
            N -= 1
            if r < half and c < half:
                continue
            elif r < half:
                result += flag
                c -= half
            elif c < half:
                result += flag * 2
                r -= half
            else:
                result += flag * 3
                r -= half
                c -= half
    return result

print(z(N, r, c))