[ 프로그래머스 ] 같은 숫자는 싫어 (Python)

기본 정보

• Level: 1
• Problem Link: 같은 숫자는 싫어
• My Solution Link: 나의 풀이
• Algorithm I used : 스택(stack)

문제

• 각 원소가 0~9의 값으로 이뤄진 배열 arr
• 연속되는 숫자는 하나만 남기기
• 출력시 arr 상의 배열 순서 유지

알고리즘

• stack…?
  • arr 원소를 왼쪽에서부터 차례대로 넣음
  • top에 있는 원소와 동일한 원소가 들어가는 경우 pop (혹은 pass)

나의 시도

def solution(arr):
    # 최종 출력할 원소를 담는 리스트
    answer = []

    for elem in arr:
        # answer 비어있나 확인
        if answer:
            # answer의 top이 elem과 동일한지 확인
            if answer[-1] != elem:
                answer.append(elem)
            else:
                continue
        else:
            answer.append(elem)

    return answer