프로그래머스 - 짝지어 제거하기

1. 문제

짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다. 먼저 문자열에서 같은 알파벳이 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을 반환합니다.

 

2. 소스코드 (Python3)

def solution(s):
    stack = []
    
    for c in s:
        if stack and stack[-1] == c:
            stack.pop()
        else:
            stack.append(c)

    return int(not stack)

 

3. 고찰

1) 자료구조

처음 이 문제를 봤을 때, 문자열을 처리할 자료구조로 리스트를 사용했다.

그런데 for문으로 인덱싱을 하면 리스트의 길이가 짧아져서 인덱싱 오류가 발생했다.

문제를 어떻게 접근해야 할지 모르겠어서 국민대학교 이시윤 교수님께 질문을 했다.

교수님께서 문자열을 차례로 살펴볼 때, 앞에서부터 순서대로 보는 것 뿐만 아니라 바로 이전 것을 확인한다고 하셨을 때 스택이구나 싶었다.

스택을 이용하면 정말 간단하게 풀 수 있는 문제였다.

 

2) 알고리즘

입력으로 받는 문자열 s의 문자들을 하나씩 살펴본다.

    만약 스택이 비어있지 않고, top이 현재 문자와 같으면 pop할 수 있다.

    여기서 파이썬 문법 : if문을 검사할 때, 왼쪽부터 검사한다고 한다. 그래서 stack이 비어있는지와 stack의 top검사를 한 줄에 쓸 수 있다.

    그렇지 않다면 stack에 push.

반복문을 수행했을 때 stack이 비어있다면 문자열을 모두 제거할 수 없으므로 0을, 비어있지 않다면 문자열을 모두 제거할 수 있으므로 1을 반환한다.

 

3) 시간복잡도

문자열 s의 길이를 N이라 할 때, 위 소스코드는 N번 for문을 반복한다.

for문안의 함수는 pop(), append() 모두 O(1)의 시간복잡도를 갖는다.

따라서 위 코드의 시간복잡도는 O(N)이다.