Programmers - 짝지어 제거하기 [Python3]

📃 문제 설명

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

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

b aa baa → bb aa → aa →

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

📌 제한사항

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

📌 입출력 예

• 입출력 예 #2
  문자열이 남아있지만 짝지어 제거할 수 있는 문자열이 더 이상 존재하지 않기 때문에 0을 반환합니다.

💡 How?

• Stack을 이용하자!
  초기 시도 때는 문자열만 이용하여 while, for 문을 돌렸으나, 코드가 복잡해질 뿐더러 시간복잡도도 늘어남을 느꼈다.
  오랜 시도 끝에도 풀리지않자, "내일 시도해보자" 하고 노트북을 덮어두었고, 산책을 하던 중에 '스택'을 이용하면 어떨까 라는 생각을 하여 곧장 집으로 돌아와서 노트북을 열었다..!

1. 스택이 비어져있을 경우, 스택에 문자를 집어넣는다.
2. 스택에 아이템이 있을 경우, 스택의 맨 마지막 원소(stack[-1])와 문자를 비교한다.
3. 일치할 경우, pop() 함수를 이용하여 스택의 마지막 원소를 제거하고, 다음 문자를 비교한다.
   (pop() 함수의 시간 복잡도는 O(1)이기 때문에 시간복잡도가 O(n)인 remove() 함수보다 더 빠른 속도로 실행할 수 있다 ✨)
4. 일치하지 않을 경우, 스택에 해당 문자를 넣고, 다음 문자를 비교한다.

+) 예외처리
주어진 문자열의 길이가 홀수일 경우, 0을 리턴한다!

👩🏻‍💻 구현한 코드

def solution(s):
    answer = 0
    stack = []

    if len(s) % 2 != 0:
        return 0

    for x in s:
        if len(stack) == 0:
            stack.append(x)
        else:
            if x == stack[-1]:
                stack.pop()
            else:
                stack.append(x)
           
    # print('------------')
    # print('s = ', s)
    # print('stack = ', stack)
   
    answer = 0 if len(stack) > 0 else 1
    return answer