볼링공 고르기

A, B두 사람이 볼링을 치고 있습니다. 두 사람은 서로 무게가 다른 골링공을 고르려고 합니다. 볼링공은 총 N개가 있으며 각 볼링공마다 무게가 적혀 이쏙, 공의 번호는 1번부터 순서대로 부여됩니다.
또한 같은 무게의 공이 여러 개 있을 수 있지만, 서로 다른 공으로 간주하빈다. 볼링공의 무게는 1부터 M까지의 자연수 형태로 존재합니다.
예를 들어 N이 5이고, M이 3이며 각각의 무게가 차례대로 1, 3, 2, 3, 2 일 때 각 공의 번호가 차례대로 1번부터 5번까지 부여됩니다. 이때 두 사람이 고를 수 있는 볼링공 번호의 조합을 구하면 다음과 같스빈다.

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

5 3
1 3 2 3 2

-> 8

8 5
1 5 4 3 2 4 5 2

-> 25

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

count = 0
for i in range(len(arr) - 1):
    for j in range(i, len(arr)):
        if arr[i] != arr[j]:
            count+=1
print("answer =", count)

해설

문제는 쉽게 해결한 것 같은데 그냥 앞 부터 차례로
다른 공을 발견하면 숫자를 더하면 된다.
굳이 인덱스를 기억할 필요도 없기 때문에 더 간단하다.

같은 무게라도 번호가 다르면 다른 공으로 생각한다고 했기 때문에, 그냥 차례로 검사하면 된다.

좋은 웹페이지 즐겨찾기