[오늘부터 코딩테스트]파이썬으로 프로그래머스 코딩 기초 프로그램 풀기

[파이썬] 프로그래머스 Lv0. n 번째 원소부터, 순서 바꾸기, 

안녕하세요 머킹입니다.

지금 자연어 처리 프로젝트의 엄청난 삽질 중이라서 오늘 제발 전처리를 끝내길 바라는 마음으로..

코딩테스트 연습을 해보겠습니다.

---

n 번째 원소부터

문제 설명

정수 리스트 num_list와 정수 n이 주어질 때, 

n 번째 원소부터 마지막 원소까지의 모든 원소를 담은 리스트를 return 하도록 solution 함수를 완성해 주세요.

 

def solution(num_list, n):
    answer = num_list[n:]
    return answer

당연히 num_list [n:]을 하는 건 줄 알았는데

테스트 1
입력값 〉	[2, 1, 6], 3
기댓값 〉	[6]
실행 결과 〉	실행한 결괏값 []이 기댓값 [6]과 다릅니다.

 

테스트 2
입력값 〉	[5, 2, 1, 7, 5], 2
기댓값 〉	[2, 1, 7, 5]
실행 결과 〉	실행한 결괏값 [1,7,5]이 기댓값 [2,1,7,5]과 다릅니다.

 

테스트 둘 다 실패했습니다 ㅎ...

def solution(num_list, n):
    if n >= len(num_list):
        return []
    answer = num_list[n:]
    return answer

그리고 두 번째 코드에서도 테스트를 다 틀렸습니다 ㅜㅜ

def solution(num_list, n):
    answer = [0] * (len(num_list) - n + 1)
    for i in range(len(answer)):
        answer[i] = num_list[n-1]
        n += 1
    return answer

결국에는 이렇게 바꿨습니다.

def solution(num_list, n):
    return num_list[n-1:]

이렇게 간단한 방법이....!!!!

---

순서 바꾸기

 

문제 설명

정수 리스트 num_list와 정수 n이 주어질 때, num_list를 n 번째 원소 이후의 원소들과 n 번째까지의 원소들로 나눠 n 번째 원소 이후의 원소들을 n 번째까지의 원소들 앞에 붙인 리스트를 return 하도록 solution 함수를 완성해 주세요.

 

def solution(num_list, n):
    answer = num_list[n:] + num_list[:n]
    return answer

위에 모범(?) 답안을 참고해서 생각해 보니까 별거 아니었습니다.

 

---

왼쪽 오른쪽

 

문제 설명

문자열 리스트 str_list에는 "u", "d", "l", "r" 네 개의 문자열이 여러 개 저장되어 있습니다. 

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

 

def solution(str_list):
    left = str_list.index('l') if "l" in str_list else float('inf')
    right = str_list.index('r') if 'r' in str_list else float('inf')
    
    if left < right:
        answer = str_list[:left]
    else:
        answer = str_list[right + 1:]
    return answer

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 []

for 문을 이용해서 하는 방법이 있네요!!