[프로그래머스 코딩테스트]Level 02 - 짝지어 제거하기
Level 02 - 짝지어 제거하기
문제 설명
짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다. 먼저 문자열에서 같은 알파벳이 2개 붙어 있는 짝을 찾습니다. 그다음, 그 둘을 제거한 뒤, 앞뒤로 문자열을 이어 붙입니다. 이 과정을 반복해서 문자열을 모두 제거한다면 짝지어 제거하기가 종료됩니다. 문자열 S가 주어졌을 때, 짝지어 제거하기를 성공적으로 수행할 수 있는지 반환하는 함수를 완성해 주세요. 성공적으로 수행할 수 있으면 1을, 아닐 경우 0을 리턴해주면 됩니다.
예를 들어, 문자열 S = baabaa 라면
b aa baa → bb aa → aa →
의 순서로 문자열을 모두 제거할 수 있으므로 1을 반환합니다.
제한사항
- 문자열의 길이 : 1,000,000이하의 자연수
- 문자열은 모두 소문자로 이루어져 있습니다.
입출력 예
s | result |
---|---|
baabaa | 1 |
cdcd | 0 |
def solution(s):
answer = 0
s = s[::-1]
s = list(s)
stack = [s.pop()]
# print(s, stack)
for word in s:
# print(stack, word)
if len(stack)==0:
stack.append(word)
elif stack[-1] != word:#스택의 탑에 있는것과 다르면 넣고
stack.append(word)
else:#같으면 뺀다
stack.pop()
# print(stack)
if not stack:
answer = 1
else:
answer = 0
return answer
문제 보는순간 아 이건 스택이다! 싶었다. 비교를 쉽게 하기 위해서 인풋으로 들어오는 문자열을 inverse해준 뒤 리스트로 변환하고, 스택이 비어있으면 처음에 비교할 수 없기 때문에 s의 top을 먼저 넣어놓고 시작한다. 이후 스택의 탑과 차례대로 비교하며 다르면 넣고 같으면 뺀다.
두 입출력 예가 동작하는 방식:
['b'] a
['b', 'a'] a
['b'] b
[] a
['a'] a
1
['c'] d
['c', 'd'] c
['c', 'd', 'c'] d
0