[이코테] 그리디 기출문제 - 5. 볼링공 고르기

---

📢 '이것이 코딩 테스트다 with 파이썬' 책을 공부하고 복습하기 위해 작성했습니다.

 

2023.12.06 - [Algorithm] - [이코테] 02. 그리디(Greedy) 알고리즘, 탐욕법, 욕심쟁이 알고리즘

[이코테] 02. 그리디(Greedy) 알고리즘, 탐욕법, 욕심쟁이 알고리즘

---

문제

A, B 두 사람은 서로 무게가 다른 볼링공으로 볼링을 치려한다.

볼링공은 총 N개가 있으며 각 볼링공마다 무게가 적혀있고, 공의 번호는 1번부터 순서대로 부여된다.

또한 같은 무게의 공이 여러 개 있을 수 있지만, 서로 다른 공으로 간주한다. 볼링공의 무게는 1부터 M까지의 자연수 형태로 존재한다.

두 사람이 볼링공을 고르는 경우의 수를 구하라

 

입력 조건

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

출력 조건

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

 

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

출력 예시 1
8

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

출력 예시 2
25

---

문제 분석 및 알고리즘 설계

2개씩 선택하며 공의 무게가 같은 지 확인한다.

공의 무게가 같다면, 고려하지 않는다.

---

나의 풀이

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

cnt = 0
for i in range(N-1):
    for j in range(i+1, N):
        if lst[i] != lst[j]:
            cnt += 1

print(cnt)

---

해설

 

효과적으로 해결하기 위해서는, 무게마다 볼링공이 몇 개 있는지 계산해야 한다.

A가 특정한 무게의 볼링공을 선택했을 때, 이어서 B가 볼링공을 선택하는 경우를 차례대로 계산하여 문제를 해결할 수 있다.

A를 기준으로 무게가 낮은 볼링공부터 무게가 높은 볼링공까지 순서대로 하나씩 확인한다.

---