Programmers, 스택, 2017 팁스타운 중 ‘짝지어 제거하기’

프로그래머스 2017 팁스타운 중 12973번 ‘짝지어 제거하기’

https://programmers.co.kr/learn/courses/30/lessons/12973

문제설명

짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다. 먼저 문자열에서 같은 알파벳이 2개 붙어 있는 짝을 찾습니다. 그다음, 그 둘을 제거한 뒤, 앞뒤로 문자열을 이어 붙입니다. 이 과정을 반복해서 문자열을 모두 제거한다면 짝지어 제거하기가 종료됩니다. 문자열 S가 주어졌을 때, 짝지어 제거하기를 성공적으로 수행할 수 있는지 반환하는 함수를 완성해 주세요. 성공적으로 수행할 수 있으면 1을, 아닐 경우 0을 리턴해주면 됩니다.

예를 들어, 문자열 S = baabaa 라면

b aa baa → bb aa → aa →

의 순서로 문자열을 모두 제거할 수 있으므로 1을 반환합니다.

제한사항

• 문자열의 길이 : 1,000,000이하의 자연수
• 문자열은 모두 소문자로 이루어져 있습니다.

입출력 예

s	result
baabaa	1
cdcd	0

입출력 예 설명

입출력 예 #1 위의 예시와 같습니다. 입출력 예 #2 문자열이 남아있지만 짝지어 제거할 수 있는 문자열이 더 이상 존재하지 않기 때문에 0을 반환합니다.

※ 공지 - 2020년 6월 8일 테스트케이스가 추가되었습니다.

---

풀이

스택의 peak 기능을 활용하면 쉽게 풀 수 있는 문제였다.

코드의 흐름은 다음과 같다.

1. 우선적으로 예외처리를 해준다. 문자열의 길이가 홀수개라면 무조건 불가능하므로 바로 0을 return 해준다.
2. 반복문을 진행하는데, 스택이 비어있다면 스택에 스캔중인 문자열의 원소를 넣는다.
3. 스택이 차 있을 때는 스택에 들어있는 가장 마지막 원소와 스캔중인 문자열의 원소가 같다면 스택에서 해당 문자를 제거해주고, 같지 않다면 스택에 집어넣는다.
4. 모든 반복을 종료하고 stk에 문자가 남아있다면 0을 return하고 비어있다면 1을 return 한다.

이를 코드로 구현하면 다음과 같다.

def solution(s):
    n = len(s)
     # 예외 처리로 조기종료
    if n % 2 != 0:
        return 0  

    stk = []
    i = 0
    while i < n:
        
         # 스택이 비었다면 스택에 원소 추가
        if not stk:
            stk.append(s[i])
            i += 1
        else:

             # 스택의 마지막 원소와 스캔중인 원소가 같다면 짝지어 문자 제거
            if stk[-1] == s[i]:
                stk.pop()
            else:
                stk += s[i]
            i += 1
            
    return 1 if not stk else 0