[프로그래머스]문제를 잘 읽자

테스트케이스 12, 19번 오류가 났던 코드.

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

 

문제를 잘 읽어야겠다.

'l'이나 'r'이 없다면 빈리스트 리턴이다.

그러므로 아래와 같이 수정한다.

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

 

성공.

if 'l' or 'r' in str_list: 이거는 안된다.

if 'l' in str_list or 'r' in str_list: 이거는 된다.

그 이유에 대해서는 다음 글에...