[알고리즘] 볼링공 고르기

*2019 SW 마에스트로 입학 문제

문제 설명

• 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번)

 

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

 

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

 

두 사람이 볼링공을 고르는 경우의 수를 출력하는 프로그램을 작성하세요

 

 

생각의 흐름

I) 예제 코드를 보니깐 (1번,2번) 이랑 (2번, 1번)이 같은 경우로 분류되는구나.

 

II) 그러면 걍 for문 돌리고 값이 같을 경우는 무시하면 되는 간단한 문제 아닌가?

 

III) 어 정말 그렇네.

 

 

코드

N, M = map(int, input().split())
lst = list(map(int,input().split()))

case = 0

for A in range(1, N):
    for B in range(A+1, N):
        if lst[A] != lst[B]:
            case += 1
print(case)

단 range부분을 좀 조심히 봐야한다. A를 기준으로 할 거니깐 A에 들어갈 값은 N-1까지이다. 물론 N을 넣는다고 문제가 되는 건 아니지만 불필요하다는 이야기이다.

 

B도 A 이전의 수를 고를 필요가 없기 때문에(중복되기 떄문에) 범위가 A의 다음 값부터 시작이다.