코딩테스트/프로그래머스

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

GAEBAL 2023. 4. 11. 16:05
728x90

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12906

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 

풀이

Level 1 문제

문제를 읽다보면 자연스럽게 스택을 생각하게 되는데, 파이썬에서는 List로 많은 자료구조를 커버할 수 있는 듯 !

 

💡 arr.append()
파이썬에는 자바와 다르게 Array와 List의 구분이 있지 않다. 파이썬의 List에서는 append() 함수를 이용해서 List의 끝에 요소를 추가할 수 있다 !
이 문제에서는 List를 Stack으로 사용하려고 함. Stack에서의 push()와 같은 기능을 한다.

 

💡arr.pop()
Stack에서의 pop()과 같은 기능을 함. 마지 요소를 pop한다 !

 

 

코드

def solution(arr):
    answer = []

    for i in arr:
        if len(answer) == 0:
            answer.append(i)
        else:
            if answer[-1] == i:
                answer.pop()
            answer.append(i)

    return answer
728x90