[Python] 볼링공 고르기

문제

• 두 사람은 서로 무게가 다른 볼링공을 고르려고 합니다. 볼링공은 총 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이 공백으로 구분되어 각각 자연수 형태로 주어집니다
• 둘째 줄에 각 볼링공의 무게 K가 공백으로 구분되어 순서대로 자연수 형태로 주어집니다

출력

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

문제 풀이 (나의 풀이)

• 숫자가 다른 경우의 조합을 찾아야 하므로 단순히 앞 뒤 값을 확인 한 후, 다른 경우 카운트
• 카운트 값 출력

코드

n, m = map(int,input().split())
data = list(map(int,input().split()))
cnt = 0
for i in range(len(data) - 1):
    for j in range(i + 1, len(data)):
        if data[i] != data[j]:
            cnt += 1

print(cnt)

문제 풀이 (책의 풀이)

• A가 무게 1인 공을 선택할 때의 경우의 수는 4가지 -> 1(무게가 1인 공의 수) X 4(B가 선택하는 경우의 수)
• A가 무게 2인 공을 선택할 때 경우의 수는 4가지 -> 2(무게가 2인 공의 수) X 2(B가 선택하는 경우의 수)
• A가 무게 3인 공을 선택할 때 경우의 수는 0가지 -> 2(무게가 3인 공의 수) X 0(B가 선택하는 경우의 수)
• 이를 모두 구한 값이 정답이므로, 공의 무게를 센 후 각 경우의 수를 result에 더한다
• 그 후 result 값 출력

코드

n, m = map(int,input().split())
data = list(map(int,input().split()))
ball_weight = [0] * 11

# 공 무게 세서 배열에 넣기
for x in data:
    ball_weight[x] += 1

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

print(result)

-> 공의 수가 많아지는 경우, 아래의 코드가 훨씬 효율적이다