문제 설명
짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다. 먼저 문자열에서 같은 알파벳이 2개 붙어 있는 짝을 찾습니다. 그다음, 그 둘을 제거한 뒤, 앞뒤로 문자열을 이어 붙입니다. 이 과정을 반복해서 문자열을 모두 제거한다면 짝지어 제거하기가 종료됩니다. 문자열 S가 주어졌을 때, 짝지어 제거하기를 성공적으로 수행할 수 있는지 반환하는 함수를 완성해 주세요. 성공적으로 수행할 수 있으면 1을, 아닐 경우 0을 리턴해주면 됩니다.
예를 들어, 문자열 S = baabaa 라면
b aa baa → bb aa → aa →
의 순서로 문자열을 모두 제거할 수 있으므로 1을 반환합니다.
제한 사항
- 문자열의 길이 : 1,000,000이하의 자연수
- 문자열은 모두 소문자로 이루어져 있습니다.
입출력 예
| s | result |
|---|---|
baabaa | 1 |
cdcd | 0 |
접근
처음에는 해당 인덱스와 다음 인덱스에 있는 요소의 값이 같은가?에 따라서 같을 때 처리를 하면 되지 않을까라는 생각을 했었다.
하지만, 이렇게 하면 계속해서 새로운 문자열을 만들어야 하는데 예외 처리 등 비효율적일 것 같은 느낌이 들었다.
그러다 현재 문자열이 다음 문자열과 같지 않다면 문자열의 맨 뒤로 보내는 생각을 하게 되었다. 다만 이때 같지 않은 문자열로만 이루어지게 되었다면 무한 반복을 하게 될텐데 이를 어떻게 해결할 수 있을까라는 고민에 빠졌다.
그러다 큐보다는 스택이 더 괜찮지 않을까라는 생각을 했다. 순차적으로 넣으면서 가장 마지막에 들어간 요소가 나랑 같으면 pop하고 아니라면 push 하면 되니까 말이다.
풀이
def solution(s):
stack = []
for char in s:
if not len(stack) or stack[-1] != char:
stack.append(char)
else:
stack.pop()
return 0 if len(stack) else 1해결 과정
stack이 비어있거나 마지막에 들어간 요소가 같지 않다면 stack에 새로운 요소를 push한다. 도중에 stack이 빌 수 있기 때문에 not len(stack) 조건은 반드시 있어야 한다.
stack의 마지막 요소가 자신과 같다면 pop을 하면 된다.
