[프로그래머스/파이썬] Lv.0 배열 만들기 4

Programmers 레벨 0_코딩 기초 트레이닝 문제 풀이

https://school.programmers.co.kr/learn/challenges/training?order=recent

코딩 기초 트레이닝 | 프로그래머스 스쿨

 

풀이 날짜 : 2023-09-13

---

문제 설명

정수 배열 arr가 주어집니다. arr를 이용해 새로운 배열 stk를 만드려고 합니다.
변수 i를 만들어 초기값을 0으로 설정한 후 i가 arr의 길이보다 작으면 다음 작업을 반복합니다.
만약 stk가 빈 배열이라면 arr[i]를 stk에 추가하고 i에 1을 더합니다.stk에 원소가 있고, stk의 마지막 원소가 arr[i]보다 작으면 arr[i]를 stk의 뒤에 추가하고 i에 1을 더합니다.stk에 원소가 있는데 stk의 마지막 원소가 arr[i]보다 크거나 같으면 stk의 마지막 원소를 stk에서 제거합니다.
위 작업을 마친 후 만들어진 stk를 return 하는 solution 함수를 완성해 주세요.

 

 

나의 풀이

def solution(arr):
    stk = []   # 빈 리스트 생성
    i = 0   # 초기식
    while i < len(arr) :            # 1. i가 arr 길이보다 작을 때까지 반복
        if stk == [] :              # 2. 조건문 1 : 빈 리스트인지 아닌지 판별
            stk.append(arr[i])
            i += 1
        else :
            if stk[-1] < arr[i] :   # 3. 조건문 2 : stk 마지막 요소가 arr[i]보다 작은지 아닌지 판별
                stk.append(arr[i])
                i += 1
            else :
                stk.remove(stk[-1])
    return stk

 

 

다른 풀이

def solution(arr):
    stk = []
    for i in range(len(arr)):
        while stk and stk[-1] >= arr[i]:
            stk.pop()
        stk.append(arr[i])
    return stk

i가 0부터 len(arr)까지 점진적으로 1씩 증가하므로 for문을 이용할 수 있다.