[문제풀이] 짝지어 제거하기

HKoiKoi·2일 전

코딩 테스트

목록 보기
12/16
post-thumbnail

문제

짝지어 제거하기

문제 설명

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

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

b aa baa →\rightarrow bb aa →\rightarrow aa →\rightarrow

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


제한사항

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

입출력 예시

sresult
baabaa1
cdcd0

문제 풀이

맨 앞의 문자부터 하나씩 스택에 넣으면서 맨 위의 문자와 넣을 문자가 같으면 제거하는 방식으로 하면 해결할 수 있을 것 같다.

baabaa를 예시로 한 번 해보자.

  1. b

스택이 비어 있으니 b를 append 한다.

스택
b
  1. a

스택이 비어 있지 않으나 스택의 제일 위 b와 a는 동일하지 않으므로 a를 append 한다.

스택
a
b
  1. a

스택이 비어있지 않고 스택의 제일 위 a와 동일하므로 스택의 a를 pop 하고 넘어간다.

스택
b
  1. b

스택이 비어있지 않고 스택의 제일 위 b와 동일하므로 스택의 b를 pop 하고 넘어간다.

스택
  1. a

스택이 비어 있으니 a를 append 한다.

스택
a
  1. a

스택이 비어있지 않고 스택의 제일 위 a와 동일하므로 스택의 a를 pop 하고 넘어간다.

스택

다 끝나고 스택의 길이가 0이면 1을, 그렇지 않으면 0을 반환하면 끝난다.

def solution(s):
    stack = []
    
    for ch in s:
        if not stack:
            stack.append(ch)
            continue
            
        if stack[-1] == ch:
            stack.pop()
            continue
            
        stack.append(ch)
            
    
    return 1 if len(stack) == 0 else 0

0개의 댓글