[python] 큰 수의 법칙

큰 수의 법칙 예제

*동빈이의 큰 수의 법칙
->다양한 수로 이루어진 배열이 있을 때 주어진 수들을 M번 더하여 가장 큰 수를 만드는 법칙. 단, 배열의 특정한 인덱스에 해당하는 수가 연속해서 K번을 초과할 수 는 없다

입력조건

• 첫째 줄에 N(2 <= N <= 1,000), M(1 <= M <= 10,000), K(1 <= K <= 10,000)의 자연수가 주어지며, 각 자연수는 공백으로 구분
• 둘째 줄에 N개의 자연수가 주어짐. 각 자연수는 공백으로 구분. 단, 각각의 자연수는 1이상 10,000이하의 수로 주어짐
• 입력으로 주어지는 K는 항상 M보다 작거나 같음

출력조건

• 첫째 줄에 동빈이의 큰 수의 법칙에 따라 더해진 답을 출력

코드

n,m,k = map(int, input().split(' '))
data = list(map(int, input().split(' ')))

data.sort()
first = data[n-1]
second = data[n-2]

result = 0

while True:
  for i in range(k):
    if m==0:
      break
    result += first
    m -= 1
  if m==0:
    break
  result += second
  m-=1

print(result)

입출력 화면