[Python] 19939번 박 터뜨리기

✏️ 문제

 

문제 파악

k = 2일 때, 최소값인 1+2보다 n이 작으면 -1

k = 3일 때, 최소값인 1+2+3보다 n이 작으면 -1 이다. 

그리고 최소값보다 클 때는 다음과 같은 규칙이 보였다.

 

알고리즘

• 그리디 알고리즘
• 수학

 

코드

n, k = map(int, input().split())
basket = k * (k + 1) // 2

if n < basket:
  print(-1)
else:
  if (n - basket) % k == 0:
    print(k - 1)
  else:
    print(k)