[COS PRO 2급 #6-3] 단체 유니폼 맞추기

문제설명

단체 유니폼을 주문하려고 합니다. 사람들의 덩치를 수치로 표현할 때 주문해야 하는 유니폼의 사이즈는 다음과 같습니다.

| 덩치              | 사이즈 |
|-------------------|--------|
| 95 미만           | S      |
| 95 이상 100 미만  | M      |
| 100 이상 105 미만 | L      |
| 105 이상          | XL     |

사람들의 덩치를 담은 리스트 people이 매개변수로 주어질 때, 주문해야 하는 유니폼 사이즈의 수를 [S, M, L, XL] 순으로 리스트에 담아 return 하도록 solution 함수를 작성해주세요.

매개변수 설명

사람들의 덩치를 담은 리스트 people이 solution 함수의 매개변수로 주어집니다.
* people의 길이는 1 이상 100 이하인 자연수입니다.
* 사람들의 덩치는 80 이상 120 이하인 자연수입니다.

return 값 설명

주문해야 하는 유니폼 사이즈의 수를 [S, M, L, XL] 순으로 리스트에 담아 return 합니다.

예시

| people                  | return       |
|-------------------------|--------------|
| [97, 102, 93, 100, 107] | [1, 1, 2, 1] |

예시설명

3번째 사람은 S 사이즈를 입어야합니다.
1번째 사람은 M 사이즈를 입어야합니다.
2, 4번째 사람은 L 사이즈를 입어야합니다.
5번째 사람은 XL 사이즈를 입어야합니다.

문제 코드

def solution(people):
    #여기에 코드를 작성해주세요.
    answer = [0 for _ in range(4)]
    return answer

풀이

def solution(people):
	answer = [0 for _ in range(4)]
	for i in people:
		if i < 95:
			answer[0] += 1
		elif i < 100:
			answer[1] += 1
		elif i < 105:
			answer[2] += 1
		else:
			answer[3] += 1
	return answer