Kasterra's PS

08/27 PS 재활 - 큰 수의 법칙

8/27/2025•5 min

어제 첫 글을 적었고, 오늘이 두번째 글이네요. 파이썬은 역시 PS 공부 할 때만 사용했었어서, PS를 오랜만에 하는 지금으로서는 파이썬 문법이 헷갈렸던지라, 구현쪽 부분에서 약간 서투른 것 같습니다. 그래도 솔직하게 이렇게 기록을 남겨보고자 합니다.

오늘 푼 문제 : 큰 수의 법칙

'큰 수의 법칙'은 일반적으로 통계분야에서 다루어지는 내용이지만, 동빈이는 본인만의 방식으로 다르게 사용하고 있다. 동빈이의 큰 수의 법칙은 다양한 수로 이루어진 배열이 있을 때, 주어진 수들을 M번 더하여 가장 큰 수를 만드는 법칙이다. 단, 이 배열의 특정한 인덱스에 해당하는 수가 연속해서 K번을 초과하여 더해질 수 없는 것이 이 법칙의 특징이다.

예를 들어 순서대로 2,4,5,4,6 으로 이루어진 배열이 있을 때, M이 8이고, K가 3이라고 가정하자. 이 경우 특정한 인덱스의 수가 연속해서 세 번까지만 더해질 수 있으므로 큰 수의 법칙에 따른 결과는 6+6+6+5+6+6+6+5 인 46이 된다.

단, 서로 다른 인덱스에 해당하는 수가 같은 경우에도 서로 다른 것으로 간주한다. 예를 들어 순서대로 3,4,3,4,3 으로 이루어진 배열이 있을 때, M이 7이고, K가 2라고 가정하자. 이 경우 두 번째 원소에 해당하는 4와, 네 번째 원소에 해다하는 4를 두 번씩 더하는 것이 가능하다. 결과적으로 4+4+4+4+4+4+4인 28이 도출된다.

배열의 크기 N, 숫자가 더해지는 횟수 M, 그리고 K가 주어질 때, 동빈이의 큰 수의 법칙에 따른 결과를 출력하시오

입력조건

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

출력조건

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

예시

입력 예시 1

5 8 3

2 4 5 4 6

출력 예시 1

46

나의 풀이

arr = []

N, M, K = map(int, input().split(' '))



temp = map(int, input().split(' '))

for i in temp:

  arr.append(i)



arr.sort(reverse=True)



acc = 0

cnt = 0

loop_cnt = 0



while cnt < M:

  if loop_cnt == K:

    acc += arr[1]

    loop_cnt = 0

  else:

    acc += arr[0]

    loop_cnt += 1

  cnt += 1



print(acc)

문제 자체는 어렵지 않습니다. 특히 인덱스가 다르면, 숫자가 같아도, 다른 것으로 간주하기에, 가장 큰 값, 그리고 두번째로 큰 값을 찾기 위해서 정렬을 한 다음, 가장 큰 수를 K번 더한 이후에, 두번째로 큰 수를 한번 더하는 것을 반복하는 것 입니다.

여기에서, 문제의 말 그대로를 코드로 직역할수 있지만, 책에서는 약간의 pharaphrase를 거친 버전을 수록하기도 하였습니다. 그 일부를 수록하겠습니다.

while True:

  for i in range(K):

    if M == 0: # M이 0이라면 반복문 탈출

      break

    result += first

    M -= 1

  if M == 0:

    break # M이 0이라면 반복문 탈출

  result += second

  M -= 1

변수를 두개를 쓰지 않고, 하나만 쓰는것도 꽤 괜찮아 보입니다.

숫자 여러개를 한 입력에 받는 법이 기억이 나지 않아 map(int, input)을 했다가, input 자체는 iterate 할 수 없다라는 사실을 에러 메세지로 알려주어, 옛 기억을 되돌아 볼 수 있게 되었습니다. 또한, map 형태가 된 입력을 list()를 이용해서 파이썬 리스트로 즉시 바꿀 수 있다는 것 또한 다시 떠올리게 되었습니다.