[문제]
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번)
결과적으로 두 사람이 공을 고르는 경우의 수는 8가지입니다. N개의 공의 무게가 각각 주어질 때, 두 사람이 볼링공을 고르는 경우의 수를 구하는 프로그램을 작성하세요.
[입력 조건]
- 첫째 줄에 볼링공의 개수 N, 공의 최대 무게 M이 공백으로 구분되어 각각 자연수 형태로 주어집니다. (1 ≤ N ≤ 1,000, 1 ≤ M ≤ 10)
- 둘째 줄에 각 볼링공의 무게 K가 공백으로 구분되어 순서대로 자연수 형태로 주어집니다. (1 ≤ K ≤ M)
[출력 조건]
- 첫째 줄에 두 사람이 볼링공을 고르는 경우의 수를 출력합니다.
<입력 예시>
5 3
1 3 2 3 2
<출력 예시>
8
[내가 푼 풀이]
🔎 아이디어
- A와 B의 무게를 순차 탐색하며, A가 고른 무게와 B가 고른 무게가 같지 않을 때 볼링공을 고르는 경우의 수를 증가시킨다.
- 번호가 1, 2, 3, 4, 5인 5개의 볼링공이 있을 때, 아래의 차례대로 [] 안에 있는 두개의 번호의 볼링공의 무게를 확인한다.
[1, 2] [1, 3] [1, 4], [1,5]
[2, 3] [2, 4] [2, 5]
[3, 4] [3, 5]
[4 5]
🔎 코드
n, m = map(int,input().split())
arr = list(map(int,input().split()))
result = 0
# 마지막은 이미 이전에 고려됨
for i in range(n-1):
# 자기자신은 고려안해도 되고, i+1 이전은 이미 이전에 고려됨
for j in range(i+1,n):
if arr[i] != arr[j]:
result += 1
print(result)
[책 풀이]
🔎 아이디어
- 무게마다 볼링공이 몇 개 있는지 계산한다.
무게가 1인 볼링공: 1개
무게가 2인 볼링공: 2개
무게가 3인 볼링공: 2개
- A를 기준으로 무게가 낮은 볼링공부터 무게가 높은 볼링공까지 순서대로 하나씩 선택한다.
- A가 특정한 무게의 볼링공을 선택하는 경우의 수와 B가 A와 다른 무게의 볼링공을 선택하는 경우의 수를 확인한다.
- B가 볼링공을 선택할 때 이미 계산 했던 경우는 제외한다.
step1> A가 무게 1인 볼링공을 선택할 때 경우의 수
1 (무게가 1인 공의 개수) × 4 (B가 선택하는 경우의 수 : 무게가 2인 볼링공 + 무게가 3인 볼링공) = 4
고려된 무게 조합: (1, 2), (1,3)
step2> A가 무게 2인 볼링공을 선택할 때 경우의 수
2 (무게가 2인 공의 개수) × 2 (B가 선택하는 경우의 수 : 무게가 3인 볼링공) = 4
여기서 무게가 (1,2) 인 조합은 이미 step1에서 계산했기 때문에, B는 무게가 1인 볼링공을 제외시킨다.
고려된 무게 조합: (2, 3)
step3> A가 무게 3인 볼링공을 선택할 때 경우의 수
2 (무게가 3인 공의 개수) × 0 (B가 선택하는 경우의 수) = 0
이전 단계에서 모든 무게 조합이 고려됬기 때문에 경우의 수는 0이 된다.
🔎 코드
n,m = map(int,input().split())
arr = list(map(int,input().split()))
# 인덱스에 해당하는 무게를 가지는 볼링공 수 저장. M은 최대 10
d = [0] * 11
for i in arr:
d[i] += 1
result = 0
for i in range(1,m+1): # a가 선택하는 무게
n -= d[i] # b가 선택하는 경우의 수. 무게가 i인 볼링공의 개수 제외, 이미 선택한건 계속 빠짐
result += d[i]*n # a가 선택하는 경우의 수 * b가 선택하는 경우의 수
print(result)
[느낀점]
N과 M같은 입력 값의 범위에 따라 효율적인 방법을 떠올리자
출처: 이것이 취업을 위한 코딩 테스트다 with 파이썬
* 포스팅에 책의 내용과 개인적인 해석이 같이 포함됩니다.
'알고리즘 문제풀이 > 파이썬' 카테고리의 다른 글
[이코테-구현] 예제 - 상하좌우 (0) | 2022.06.17 |
---|---|
[이코테-그리디] 기출 - 06.무지의 먹방 라이브 (0) | 2022.05.09 |
[이코테-그리디] 기출 - 04. 만들 수 없는 금액 (1) | 2022.05.08 |
[이코테-그리디] 기출 - 03. 문자열 뒤집기 (0) | 2022.05.06 |
[이코테-그리디] 기출 - 02. 곱하기 혹은 더하기 (0) | 2022.05.06 |