Python/알고리즘

스택/큐 Level 1 같은 숫자는 싫어 - python (프로그래머스)

구름용 2023. 4. 30. 16:42

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

 

프로그래머스

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

programmers.co.kr

 

소스 코드


def solution(arr):
    
    result = []
    

    for value in arr:
        if len(result) == 0:
            result.append(value)
        else:
            if result[len(result) - 1] != value:
                result.append(value)
    
    
    
    return result

 

풀이

문제의 핵심은 result 리스트에 두 번 연속으로 같은 값이 들어오지 않도록 하는 것이다.

따라서 result 리스트에 값을 push 하는 경우는 2가지가 있다.

 

1. result 배열이 비어있을 경우

  • 배열이 비어있을 경우 절대로 같은 값이 들어올 수 없다.

2. result 배열의 마지막 값이 value와 같지 않을 경우

 

따라서 처음에는 1. 경우와 2. 경우를 if else로 나누어 최종적으로는 2중 조건문을 사용하였다. 하지만 추후에 하나의 if로 표현 가능할 수 있음을 깨달아서 다음과 같이 개선하였다.

def solution(arr):
    
    result = []
    

    for value in arr:
        if len(result) == 0 or result[len(result)-1] != value:
            result.append(value)
    
    
    return result

 

다른 사람의 풀이

다른 사람의 풀이 역시 큰 틀은 바뀌지 않았으며 list comprehension이나 주로 len 함수를 사용하지 않고 리스트 슬라이싱을 이용한 것을 볼 수 있었다.