지원연구소 공식 블로그

[PROGRAMMERS] 왼쪽 오른쪽 (LV. 0) - Python 본문

코딩테스트/프로그래머스

[PROGRAMMERS] 왼쪽 오른쪽 (LV. 0) - Python

지원연구소 2024. 5. 2. 00:41

[ 문제 ]

문자열 리스트 str_list에는 "u", "d", "l", "r" 네 개의 문자열이 여러 개 저장되어 있습니다. str_list에서 "l"과 "r" 중 먼저 나오는 문자열이 "l"이라면 해당 문자열을 기준으로 왼쪽에 있는 문자열들을 순서대로 담은 리스트를, 먼저 나오는 문자열이 "r"이라면 해당 문자열을 기준으로 오른쪽에 있는 문자열들을 순서대로 담은 리스트를 return하도록 solution 함수를 완성해주세요. "l"이나 "r"이 없다면 빈 리스트를 return합니다.

 

 

 

[ 제한 사항 ]

  • 1 ≤ str_list의 길이 ≤ 20
  • str_list는 "u", "d", "l", "r" 네 개의 문자열로 이루어져 있습니다.

 

 

[ 입출력 예 ]

str_list result
["u", "u", "l", "r"] ["u", "u"]
["l"] []

 

 

 

[ 해결 방법 ]

1. str_list 길이로 반복
2. 'l'과 "r" 중에 먼저 나오는 값이 "l"이면 해당 위치 기준 왼쪽을, "r"이면 오른쪽을 반환
3. "l"과 "r"이 포함되지 않은 리스트라면 빈 리스트를 반환한다.

 

 

 

[ solution.py ]

def solution(str_list):
    for i in range(len(str_list)):
        if str_list[i] == 'l':
            return str_list[ : i]
        elif str_list[i] == 'r':
            return str_list[i+1 : ]
        
    return []

 

 

 

> 🌻피드백은 언제나 환영입니다🌻 <