알고리즘/파이썬

[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, "입니다.")
반응형