알고리즘/파이썬
[COS Pro 2급, Python] 1차 1번 : 단체 티셔츠 주문하기
Internal Server Error
2024. 1. 2. 20:06
반응형
문제
더보기
A 학교에서는 단체 티셔츠를 주문하기 위해 학생별로 원하는 티셔츠 사이즈를 조사했습니다. 선택할 수 있는 티셔츠 사이즈는 작은 순서대로 "XS", "S", "M", "L", "XL", "XXL" 총 6종류가 있습니다.
학생별로 원하는 티셔츠 사이즈를 조사한 결과가 들어있는 배열 shirtsize와 shirtsize의 길이 shirtsizelen이 매개변수로 주어질 때, 사이즈별로 티셔츠가 몇 벌씩 필요한지 가장 작은 사이즈부터 순서대로 배열에 담아 return 하도록 solution 함수를 완성해주세요.
Code
def solution(shirt_size):
answer = [0, 0, 0, 0, 0, 0]
for i in range(len(shirt_size)) :
if shirt_size[i] == "XS" :
answer[0] += 1
elif shirt_size[i] == "S" :
answer[1] += 1
elif shirt_size[i] == "M" :
answer[2] += 1
elif shirt_size[i] == "L" :
answer[3] += 1
elif shirt_size[i] == "XL" :
answer[4] += 1
else :
answer[5] += 1
return answer
shirt_size = ["XS", "S", "L", "L", "XL", "S"]
ret = solution(shirt_size);
print("solution 함수의 반환 값은", ret, "입니다.")
반응형