[이코테 Greedy] 볼링공 고르기

# 문제

A, B 두 사람이 볼링을 치고 있습니다. 두 사람은 서로 무게가 다른 볼링공을 고르려고 합니다. 볼링공은 총 N개가 있으며 각 볼링공마다 무게가 적혀 있고, 공의 번호는 1번부터 순서대로 부여됩니다. 또한 같은 무게의 공이 여러개 있을 수 있지만, 서로 다른 공으로 간주합니다. 볼링공의 무게는 1부터 M까지의 자연수 형태로 존재합니다.

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

# 입력 조건

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

# 출력 조건

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

# 입력 예시

# 입력 예시 1
5 3
1 3 2 3 2

# 입력 예시 2
8 5
1 5 4 3 2 4 5 2

# 출력 예시

# 출력 예시 1
8
# 출력 예시 2
25

 

---

내 풀이

from itertools import combinations

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

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

result = 0  # 경우의 수

# N개 중 2개 뽑는 모든 경우의 수 구하기
for i in combinations(balls, 2):
    result += 1

# 무게 1 ~ M 각각에 대하여 같은 무게의 공이 몇 개인지 세기
for weight in range(1, m + 1):
    same_num = balls.count(weight)
    # 같은 무게의 공이 없을 경우
    if same_num <= 1:
        continue
    # 같은 무게의 공이 있을 경우, 해당 무게의 공 중 2개를 뽑는 경우의 수를 구해서 전체 경우의 수에서 빼기
    for i in combinations(range(same_num), 2):
        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)

 

주목해볼 점

 조합의 수를 구할 때 for문으로 센 점을 기억하자. 지향할 답안 깔끔하니 참고하자.