알고리즘/파이썬
[COS Pro 2급, Python] 6차 3번 : 단체 유니폼 맞추기
Internal Server Error
2024. 1. 12. 18:26
반응형
문제
단체 유니폼을 주문하려고 합니다. 사람들의 덩치를 수치로 표현할 때 주문해야 하는 유니폼의 사이즈는 다음과 같습니다.
덩치 | 사이즈 |
95 미만 | S |
95 이상 100 미만 | M |
100 이상 105 미만 | L |
105 이상 | XL |
사람들의 덩치를 담은 배열 people, people의 길이 people_len이 매개변수로 주어질 때, 주문해야 하는 유니폼 사이즈의 수를 [S, M, L, XL] 순으로 배열에 담아 return 하도록 solution 함수를 작성해주세요.
code
def solution(people):
answer = [0 for _ in range(4)]
uni_size = ["S", "M", "L", "XL"]
for i in people :
if i < 95 :
answer[0] += 1
elif i >= 95 and i < 100 :
answer[1] += 1
elif i >= 100 and i < 105 :
answer[2] += 1
elif i >= 105 :
answer[3] += 1
return answer
people = [97, 102, 93, 100, 107]
ret = solution(people)
print("solution 함수의 반환 값은", ret, "입니다.")
반응형