[프로그래머스/Python] 짝지어 제거하기

문제 정보

제목 : 짝지어 제거하기

난이도 : Lv.2

사용 언어 : Python

문제 링크

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

 

문제 설명

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

 

제한 사항

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

입출력 예

s	result
baabaa	1
cdcd	0

입출력 예 설명

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

---

나의 풀이

스택으로 문제를 풀면 간단하게 풀립니다. 문자열 s의 길이만큼 반복하며 리스트 stack을 만든후 첫 문자는 stack에 추가합니다. 그후 stack의 마지막 인덱스의 문자와 i가 일치하다면 stack에서 pop()을 해주고 일치하지 않다면 append()해줍니다. 반복문이 끝나면 if문을 이용하여 stack의 길이가 0이면 answer에 1을 저장하고 0이 아니라면 0을 저장하여 리턴합니다.

 

코드

def solution(s):
    answer = -1
    stack = []
    
    for i in s :
        if len(stack) == 0 :
            stack.append(i)
        else :
            if stack[-1] == i :
                stack.pop()
            else :
                stack.append(i)
                
    if len(stack) == 0 :
        answer = 1
    else:
        answer = 0

    return answer