[CosPro 2급 1차] 1번 정답 및 해설

edu.goorm.io/learn/lecture/17033/cos-pro-2%EA%B8%89-%EA%B8%B0%EC%B6%9C%EB%AC%B8%EC%A0%9C-python

구름HOME

 

 

 

🙌  문제 설명

해당 문제는 shirt_size 리스트를 입력받고, 각 사이즈별로 몇 개가 있는지를 구하는 문제입니다.

이해를 돕기 위해 문제에서 제시된 예시가 아닌 다른 예를 들어볼게요.

 

[ 매개변수 설명 ]

매개변수는 어떤 함수를 호출할 때 넘겨주는 값이예요.

이 문제에서는 매개변수로 shirt_size 라는 리스트를 받아요.

여기에는 무작위 순서의 옷 사이즈들이 저장되어 있어요.

 

[ return 값 설명 ]

return 값은 함수가 호출되어 코드를 실행하고 난 뒤, 남기는 값이예요.

여기에서는 각 사이즈별 개수를 저장해서 남기라고 하고 있어요

 

 

만약, shirt_size 리스트에 ["S", "L", "S", "XXL", "XS", "L"]가 있다면, 다음과 같이 정리할 수 있어요.

XS : 1개
S : 2개
M : 0개
L : 2개
XL : 0개
XXL : 1개

 

그럼, 여기에서 오른쪽에 있는 개수만 리스트에 담아 보여주면 문제를 해결할 수 있어요.

[1, 2, 0, 2, 0, 1] 과 같은 형태의 리스트를 반환(return) 해주면 돼요.

 

---

 

 

그럼 해당 문제를 풀기 위한 순서들을 말로 이해하기 쉽게 정리해볼게요.

1. 각 사이즈의 개수를 저장할 공간을 만든다.

2. shirt_size에 있는 사이즈를 하나씩 가져온다.

3. 가져온 사이즈가 어떤 사이즈인지 확인한다.

--> 만약, 가져온 사이즈가 S라면 리스트의 두번째 요소. 즉, 인덱스 1인 자리에 갯수를 하나 더해주면 돼요.

--> 리스트의 인덱스는 0부터 시작하므로, 두번째 칸이지만, 인덱스 값은 1이예요 !

4. 결과가 담긴 리스트를 반환(return)한다.

 

그럼 이번에는, 이렇게 말로 적은 순서들을 코드와 관련지어 차근차근 설명해볼게요.

 

1. 각 사이즈의 개수를 저장할 공간을 만든다.

코드를 짤 때는 필요한 변수들을 가장 먼저 선언해주는 것이 좋아요. 

 

우리는 각 사이즈별로 분류하고, 해당 사이즈가 총 몇개 필요한지 구해야 하므로, 각 사이즈 별 서랍을 만들거예요.

서랍의 순서는 XS, S, M, L, XL, XXL 의 순서로 나열되어 있어요.

이 때, 인덱스는 0부터 시작하므로, S 사이즈인 옷을 만났다면, 인덱스가 1인 서랍에 1을 더해주면 돼요.

그럼, 각 공간을 만들고, 아직은 아무것도 세지 않았으므로 개수를 모두 0으로 설정해줄게요.

answer = [0, 0, 0, 0, 0, 0]
       # XS, S, M, L, XL, XXL

 

 

2. shirt_size에 있는 사이즈를 하나씩 가져온다.

이를 수행하려면, shirt_size라는 리스트에 있는 요소들을 하나씩 가져와야 해요.

이렇게 하나씩 가져오려면, 우리는 '반복'을 사용할 수 있어요.

'반복'문은 크게 for문과 while문이 있는데요, 위 문제와 같은 경우, size_shirt에 있는 것들을 모두 확인하고 나면 끝내도 되므로 종료 지점이 명확해요.

이처럼 종료 지점이 명확한 경우에는 for문을 쓰는 것이 좋아요.

여기에서 종료지점은 size_shirt에 내용 갯수만큼 보면 되니, len() 함수를 사용하면 돼요.

 

그럼, shirt_size에 있는 사이즈를 하나씩 가져오는 코드를 한 번 적어볼게요 !

for i in range(len(shirt_size)) :
	print(shirt_size[i]) # 설명을 위한 코드이므로 적지 않아도 돼요!

이렇게 하면, 받은 리스트에 있는 내용에 하나씩 접근해서 출력할 수 있어요.

for문은 한 번 반복할 때마다 자동으로 다음으로 넘어가요 !

 

 

3. 가져온 사이즈가 어떤 사이즈인지 확인한다.

두번째로 이렇게 for문으로 하나씩 가져온 값을 확인해야 해요.

가져온 옷의 크기가 S라면, 아까 처음에 만든 answer 리스트의 1번째에 1을 추가해줘야겠죠?

(S는 두 번째 서랍인데, 인덱스는 0부터 시작하므로, 인덱스 1인 자리에 1을 더해주면 돼요.)

 

S를 예시로, 코드를 한 번 짜볼게요.

for i in range(len(shirt_size)) :
   if shirt_size[i] == 'S' :
    	answer[1] += 1

이렇게 가져온 shirt_size가 S라면, 1번째 값을 1 더해주면 돼요 !

그럼 나머지도 적어볼게요.

for size in shirt_size:
        if size == "XS":
            answer[0] += 1
        elif size == "S":
            answer[1] += 1
        elif size == "M":
            answer[2] += 1
        elif size == "L":
            answer[3] += 1
        elif size == "XL":
            answer[4] += 1
        else:
            answer[5] += 1

 

이렇게 하면, 최종 코드를 완성할 수 있어요.

지금까지 순서대로 한 코드를 모두 합쳐볼게요.

 

1. 각 사이즈의 개수를 저장할 공간을 만든다.

2. shirt_size에 있는 사이즈를 하나씩 가져온다.

3. 가져온 사이즈가 어떤 사이즈인지 확인한다.

4. 결과가 담긴 리스트를 반환(return) 한다.

 

 

정답 코드

def solution(shirt_size):
    answer = [0, 0, 0, 0, 0, 0]
    for size in shirt_size:
        if size == "XS":
            answer[0] += 1
        elif size == "S":
            answer[1] += 1
        elif size == "M":
            answer[2] += 1
        elif size == "L":
            answer[3] += 1
        elif size == "XL":
            answer[4] += 1
        else:
            answer[5] += 1
    return answer