[11-05 그리디 문제] 볼링공 고르기

최근 보고 있는 책인  '이것이 코딩테스트다 with 파이썬 편_나동빈_한빛미디어' 에 있는 문제이다.

www.hanbit.co.kr/store/books/look.php?p_code=B8945183661

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

 

 

문제는 책 안에 있다!! 

import sys
input = sys.stdin.readline
n,m = map(int,input().split())
temp = list(map(int,input().split()))

weight = [0]*(m+1)

for i in temp :
    weight[i] += 1

result = 0
for i in range(1,m):
    if weight[i] > 0 :
        result += (weight[i]*sum(weight[i+1:]))

print(result)

 

이 문제를 해결하면서, 처음 딱 봤을 때 문제에 대한 실마리가 보이지 않는 경우, 예시로 유추하여 해결한 후, 예외 조건이 있는지 확인하는 것도 좋다는 것을 알 수 있었다.