[탐욕법] 난이도1, 2019 SW 마에스트로 입학 테스트 '볼링공 고르기' (Python)

<문제>

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. 두 사람이 고를 수 있는 볼링공 번호의 조합을 구해야 하므로 조합 라이브러리인 combinations 사용
2. 두 사람은 서로 무게가 다른 볼링공을 골라야 하므로, 만든 조합의 개수에서 무게가 같은 볼링공의 개수를 뺀다.

<코드>

from itertools import combinations

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

com = list(combinations(k,2))
result = len(com)

for i in com:
	if i[0]==i[1]:
    	result -= 1
print(result)

#220321
from itertools import combinations

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

ball = list(map(int,input().split()))

answer = 0
for a, b in list(combinations(ball, 2)):
    if a != b:
        answer += 1

print(answer)

 

#라이브러리 안 쓰는 코드
n,m = map(int, input().split())
data = list(map(int, input().split()))

array = [0]*11

for x in data:
	array[x] += 1
    
result = 0
for i in range(1,m+1):
	n-=array[i]
    result += array[i]*n

print(result)

 

<고쳐야 할 점>

• permutations와 combinations 차이점 제대로 알기