짝지어 제거하기(python)

참고 자료

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

문제 설명

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

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

b aa baa → bb aa → aa →

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

제한사항

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

입출력 예

arrreturn
baabaa1
cdcd0

나의 생각

문자열 압축 문제가 생각이 나서 for문과 slicing을 통해서 진행하려고 했지만 시간 초과가 났다.
제한사항의 문자열의 길이 제한으로 시간 초과가 날 것 같다는 생각은 했다. stack을 활용할 생각을 못했다.

코드

  • slicing 활용.(시간 초과)
def solution(s):
    answer = 0
    issame = True
    while issame:
        issame = False
        cnt = 0
        s_idx = 0
        e_idx = 1
        for a, b in zip(s, s[1:] + " "):
            if a == b:
                e_idx += 1
                issame = True
                cnt += 1
            else:
                if cnt > 0:
                    s = s[:s_idx] + s[e_idx : ]
                    break
                s_idx = e_idx
                e_idx += 1
    if s:
        answer = 0
    else:
        answer = 1
    return answer
  • stack을 활용
def solution(s):
    st = []
    for i in s:
        if len(st) == 0:
            st.append(i)
        else:
            if st[-1] == i:
                st.pop()
            else:
                st.append(i)
    if not st:
        return 1
    else:
        return 0