백준 1052 [물병] 파이썬

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

1052번: 물병

import sys
sys.setrecursionlimit(1000000) //재귀 제한 범위 확장
N,K=map(int,sys.stdin.readline().split()) //N,K 입력받음


def check(n,k):
  val=1
  while val<n: // 2의 제곱이 n보다 커질때까지 반복
    val*=2
  if k==1: // k가 1이면 val과 n의 차 만큼 물병이 더 필요
    return val-n
  else: // k가 1이 아니라면 
    val//=2 // val<n으로 변경
    return check(n-val,k-1) // n-val개의 물병으로 k-1만큼 물병을 만드는 것으로 생각가능


if N<K:
  print(0) 
else:
  print(check(N,K))