[그리디 알고리즘] 볼링공 고르기 - 파이썬(python)

볼링공 고르기

난이도 : 下 풀이 시간 : 30분

시간 제한 : 1초 메모리 제한 : 128 MB

 

---

 

해답

 

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

array = [0] * 11
count = 0

for i in k:
    array[i] = array[i] + 1

for i in range(1, m+1):
    n = n - array[i]
    count = count + (array[i] * n)

print(count)

---

예시

 

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)

---

해설

 

A가 특정 무게를 선택하였을 때 B가 선택하는 경우를 차례대로 계산하면 됩니다.