이것이코딩테스트다 | Q05 볼링공 고르기

Date | 2021.07.23

✅ 난이도 _ 🌝🌑🌑
✅ 풀이 시간 _ 30분
✅ 시간 제한 _ 1초
✅ 메모리 제한 _ 128MB
✅ 기출 _ 2019 SW 마에스트로 입학테스트

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

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

 

---

 

👩‍🔧 코드

### 이것이 코딩테스트다 p.315###
### Q 05 - 볼링공 고르기 ###

# 12:17 ~ 12:50 (33m)

# m, n 입력받기
n, m = map(int, input().split())

# 무게 입력 받기
k = list(map(int, input().split()))

# 무게 별 공 개수 리스트로 정리
ball = [0] * m
for i in range(n):
  w = k[i]
  ball[w-1] += 1

result = 0 # 경우의 수

for a in range(0,m):
  for b in range(a+1,m):
    result += ball[a]*ball[b]

print(result)

 

 

👮‍♀️ 실행 결과

 

👩‍🏫 알고리즘 노트