Python 20제] 2020년도 한국정보올림피아드 1차 대회 초등부/고등부 2교시 문제1. 박 터뜨리기

출처 : 반크 카드뉴스

 

 

문제]

K개의 팀이 박터트리기 게임을 한다. 각 팀은 하나의 바구니를 가지고 있고 바구니에 들어있는 공들을 던겨서 자기 팀의 박을 터드려야 한다.

우리는 게임을 준비하기 위해서, N개의 공들을 K개의 바구니에 나눠 담아야 한다. 이 때, 게임의 재미를 위해서 바구니에 담기는 공들의 개수를 모두 다르게 하고 싶다. 다시 말해서, N개의 공을 K개의 바구기에 빠짐없이 나누어 담는데, 각 바구니에는 1개 이상의 공이 있어야 하고, 바구니에 담긴 공들의 개수가 모두 달라야 한다.
더불어, 게임의 불공정함을 줄이기 위해서, 가장 많이 담기 바구니와 가장 적게 담긴 바구니의 공의 차이가 최소가 되도록 담을 것이다.
공들을 바구에 나눠 담기 위한 규칙들을 정리하면 다음과 같다.
1) N개의 공을 K개의 바구니에 빠짐없이 나누어 담는다.
2) 각 바구니에는 1개 이상의 공이 들어 있어야 한다.
3) 각 바구니에 담기 공들의 개수는 모두 달라야 한다.
4) 가장 많이 담긴 바구니와 가장 적게 담기 바구니의 공의 개수 차이가 최소가 되어야 한다.
위 규칙들을 만족하며 N개의 공들을 K개의 바구니에 나눠 담을 때, 나눠 담을 수 있는 지 여부를 결정하고, 담을 수 있을 때, 가장 많이 담긴 바구니와 가장 적게 담긴 바구니의 공의 개수 차이를 계산해서 출력하는 프로그램을 작성하시오.

입력 형식]
표준 입력으로 다음 정보가 주어진다. 첫 번째 줄에 공들의 개수를 나타내는 N과(2≤N≤100,000) 팀 수를 나타내는 정수 K(2≤K≤1,000)가 주어진다.

출력 형식]
표준 출력으로 하나의 정수를 다음과 같이 출력하라.
·N개의 공들을 K개의 바구니에 문제의 규칙을 맞곡하면서 나눠 담을 수 있다면, 가장 많이 담긴 바구니와 가장 적게 담긴 바구니의 공의 개수 차이를 출력하라.
·N개의 공들을 K개의 바구니에 문제의 규칙을 만족하면서 나눠 담을 수 없다면, -1을 출력한다.


예시 1]
입력
5 3

출력
-1


예시 2]
입력
6 3

출력
2

 

 

출처]

2020년 한국정보올림피아드 1차대회 2교시 초등부/고등부pdf

 

 

참고풀이]

#공개수와 팀수를 입력받는다.
N, K = map(int,input().split(' '))

#공의 개수와 팀수가 조건에 맞는 지 체크
if (2<=N<=100000) and (2<=K<=1000):
    N-=K*(K+1)//2
    if N<0:
        print("-1")
    else:
        print(K-1 if N%K == 0 else K)

 

 

참고풀이 결과]

 

 

 

대한민국의 아름다운 영토, 독도의 여름