(Python)백준 코딩테스트 연습 - 물병(1052)

1. 문제

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

1052번: 물병

2. 풀이

import sys
input = sys.stdin.readline

# n : 물병 수, k : 옮길 수 있는 물병 수
n, k = map(int, input().split())
if n <= k:
    print(0)
else:
    while k != 1 and n != 1:

        squareNum = 1
        while squareNum <= n:
            squareNum *= 2
        squareNum //= 2
#         print("squareNum :", squareNum)
        n -= squareNum
        k -= 1
#     print("n :", n)
    if n == 0:
        n += 1
    squareNum = 1
    while squareNum < n:
        squareNum *= 2
    print(squareNum - n)