[Greedy] 이코테 - 볼링공 고르기(py)

문제

A, B 두 사람이 볼링을 치고 있습니다. 두 사람은 서로 무게가 다른 볼링공을 고르려고 합니다. 볼링공은 총 N개가 있으며 각 볼링공마다 무게가 적혀 있고, 공의 번호는 1번부터 순서대로 부여됩니다. 또한 같은 무게의 공이 여러 개 있을 수 있지만, 서로 다른 공으로 간주합니다. 볼링공의 무게는 1부터 M까지의 자연수 형태로 존재합니다.
예를 들어 N이 5이고, M이 3이며 각각의 무게가 차례대로 1,3,2,3,2일 때 각 공의 번호가 차례대로 1번부터 5번까지 부여됩니다. 이때 두 사람이 고를 수 있는 볼링공 번호의 조합을 구하면 다음과 같습니다.

(1번, 2번), (1번, 3번), (1번, 4번), (1번, 5번), (2번, 3번), (2번, 5번), (3번, 4번), (4번, 5번)

결과적으로 두 사람이 공을 고르는 경우의 수는 8가지입니다. N개의 공의 무게가 각각 주어질 때, 두 사람이 볼링공을 고르는 경우의 수를 구하는 프로그램을 작성하세요.

 

입력 조건

• 첫째 줄에 볼링공의 개수 N, 공의 최대 무게 M이 공백으로 구분되어 각각 자연수 형태로 주어집니다. (1 <= N <= 1,000 ,1 <= M <= 10)
• 둘째 줄에 각 볼링공의 무게 K가 공백으로 구분되어 순서대로 자연수 형태로 주어집니다. (1 <= K <=M)

출력 조건

• 첫째 줄에 두 사람이 볼링공을 고르는 경우의 수를 출력합니다.

입력 예시	출력 예시
5 3 1 3 2 3 2	8
8 5 1 5 4 3 2 4 5 2	25

---

풀이

- 내 풀이

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

cnt = 0
for i in range(0, len(k)-1):
  target = k[i]
  for j in range(i+1, len(k)):
    if target != k[j]:
      cnt += 1

print(cnt)

이중 반복문을 이용해 처음부터 하나씩 비교해 서로 다른 무게일 경우의 수를 세준다.

 

- 답안

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

# 1 ~ 10 무게를 담을 수 있는 리스트
array = [0] * 11

# 각 무게에 해당하는 볼링공의 개수 카운트
for x in data:
  array[x] += 1

result = 0

# 1부터 m까지의 각 무게에 대하여 처리
for i in range(1, m + 1):
  n -= array[i]  # 무게가 i인 볼링공의 개수(A가 선택할 수 있는 개수) 제외
  result += array[i] * n  # B가 선택하는 경우의 수와 곱하기

print(result)

 

이중 for문을 돌리지 않는 답안이 더 효율적

 

첫 번째 입출력 예시에서

• 무게가 1인 볼링공: 1개
• 무게가 2인 볼링공: 2개
• 무게가 3인 볼링공: 2개

step1) A가 무게가 1인 볼링공 선택

: 1(무게 1인 공의 개수) x 4(B가 선택하는 경우의 수) = 4가지

 

step2) A가 무게가 2인 볼링공 선택

: 2(무게가 2인 공의 개수) x 2(B가 선택하는 경우의 수) = 4가지

 

step3) A가 무게가 3인 볼링공 선택

: 2(무게가 3인 공의 개수) x 0 (B가 선택하는 경우의 수) = 0가지

 

따라서 가능한 경우의 수는 4 + 4 + 0 = 총 8가지

 

아이디어

• 먼저 무게마다 볼링공이 몇 개 있는지를 계산하고
• A가 특정한 무게의 볼링공을 선택했을 때, 이어서 B가 볼링공을 선택하는 경우를 차례대로 계산하여 문제를 해결
• A를 기준으로 무게가 낮은 볼링공부터 무게가 높은 볼링공까지 순서대로 하나씩 확인한다고 했을 때
• 단계(step)가 진행됨에 따라, 'B가 선택하는 경우의 수'는 줄어든다. => 이미 계산했던 경우(조합)는 제외하기 때문
• 볼링공의 무게가 1 ~ 10만 존재할 수 있다고 명시되어 있기 때문에 => 하나의 리스트 변수 선언, 각 무게별로 볼링공이 몇 개가 존재하는지 기록할 수 있음

 

+ 리스트를 만들고 각 인덱스에 해당하는 수를 세는 코드 알고 있자!

위 코드에서 list가 0부터 시작하기 때문에 'array = [0] * 11'로 선언해 주었다. 10으로 선언하고 아래 코드를 같이 수정해도 같은 답이 나오지만, 리스트 할당 시 c++을 배울 때 조금 여유를 두는 것도 좋다고 알고 있다. 위 그림은 답안 코드를 pythontutor.com에서 돌려보면 볼 수 있다(너무 편하다).

---

 

출처: 나동빈, 「이것이 취업을 위한 코딩테스트다 with 파이썬」, 한빛미디어, p.315