[Python] 이것이 코딩테스트다 with 파이썬 - 볼링공 고르기 - 그리디 알고리즘

해당 문제 및 해설은 "이것이 코딩테스트 다 with 파이썬" 책을 기준으로 요약 및 정리하여 작성하였습니다.

 

이것은 코딩테스트다 with 파이썬 구입처

이것이 취업을 위한 코딩 테스트다 with 파이썬 - YES24

---

문제

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 )

 

출력 조건

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

 

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

 

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

 

내 코드 -   테스트 케이스 통과

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

data = list(map(int, input().split()))
result = 0

for i in range(n - 1):
  for j in range(i+ 1, n):
    if data[i] == data[j]:
      continue
    result += 1

print(result)

 

답안 코드 - 책, 테스트 케이스 통과

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)