[프로그래머스] 짝지어 제거하기 - 스택 (python)

1. 문제 설명

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

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

 

b aa baa → bb aa → aa →

 

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

 

---

2. 제한 사항

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

 

---

3. 입출력 예

s	result
baabaa	1
cdcd	0

 

---

4. 나의 풀이

이 문제는 스택을 써서 간단하게 풀 수 있다. 

 

def solution(s):
    if len(s)%2 == 1:
        return 0
    stack = [s[0]]
    
    for i in s[1:]:
        if len(stack) == 0 or stack[-1] != i:
            stack.append(i)
        else:
            stack.pop()

    return 1 if len(stack) == 0 else 0

 

처음에는 혹시나, 효율성에 영향이 있을까봐 문자열 길이가 홀수이면 실패를 리턴하는 조건문을 추가해주었다. (별로 의미는 없는 것 같다.)

다음부턴 스택에 첫 글자를 넣고 반복문을 이용해 모든 글자를 아래처럼 판별하였다.

1) 스택에 아무 글자도 없거나, 스택 맨 위의 글자가 넣을 차례의 글자와 다르면 스택에 추가해준다.

2) 스택 마지막 글자가 넣을 차례의 글자와 같으면 스택의 마지막 요소를 pop해준다.

 

모든 과정이 끝나고 나서 스택을 판별했을 때 스택이 비었으면 1(성공)을, 아니면 0(실패)를 출력해주면 끝이다.

 

문제를 처음 보고는, 글자를 하나하나 보면서 겹치는 부분이 있으면 제거하고, while문을 이용해 다시 처음으로 돌아가 판별하는 방식으로 하려고 했는데 효율성에서 뻑이났다. 스택을 사용하니 O(n)으로 문제없이 돌아갔다. 간단한 문제를 복잡하게 생각하다가 시간을 많이 잡아먹어서 이런 부분에서 훈련이 많이 필요하다는 것을 느꼈다.

 

 

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

 

코딩테스트 연습 - 짝지어 제거하기