왼쪽 오른쪽

송용진·2023년 11월 4일
0

알고리즘

목록 보기
149/173

문자열 리스트 str_list에는
"u", "d", "l", "r" 네 개의 문자열이 여러 개 저장되어 있습니다.
str_list에서 "l"과 "r" 중
먼저 나오는 문자열이 "l"이라면
해당 문자열을 기준으로
왼쪽에 있는 문자열들을 순서대로 담은 리스트를,

먼저 나오는 문자열이 "r"이라면
해당 문자열을 기준으로
오른쪽에 있는 문자열들을 순서대로 담은 리스트를

return하도록 solution 함수를 완성해주세요.
"l"이나 "r"이 없다면 빈 리스트를 return합니다.

내 코드

def solution(str_list):
    answer = []
    for i,v in enumerate(str_list):
        if not(v == "l" or v == "r"):
            continue
        elif v == "l":
            answer = str_list[:i]
            break
        elif v == "r":            
            answer = str_list[i+1:]
            break
    return answer

print(solution(["u", "u", "l", "r"]))
print(solution(["l"]))

예시 코드

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 []
profile
백엔드 개발자

0개의 댓글