[프로그래머스][Lv. 2][Python] 짝지어 제거하기

https://school.programmers.co.kr/learn/courses/30/lessons/12973

프로그래머스

✔ 문제 설명

 

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

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

b aa baa → bb aa → aa →

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

제한사항

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

---

 

▪ Before (런타임 에러)

 

1.  i가 s길이 만큼 진행될때 까지 반복한다.
2. s[i]와 s[i+1]이 같으면 해당 부분을 제외한 s를 새로 정의하고 i를 max(0, i-1)로 바꾼다.(i가 0 일시 i-1은 오류가 나기 때문)
3. s의 길이가 0일시 1을 출력하고 길이가 남으면 0을 출력한다.

 

def solution(s):
    i = 0
    while i < len(s) - 1:
        if s[i] == s[i+1]:
            s = s[0:i] + s[i+2:]
            i = max(0, i-1)
        else:
            i += 1
        
        
    if len(s) == 0:
        return 1
    else: 
        return 0

▪ After (stack)

 

1. stack이 비어 있을 시 해당 문자를 stack에 쌓는다.
2. stack이 비어 있지 않은 상태에서 해당 문자가 stack 마지막에 들어온 문자와 같으면 stack에 존재하는 마지막 문자를 제거한다.
3. 같지 않으면 stack에 쌓는다.
4. s의 모든 문자를 반복문으로 확인하였을 때 stack에 쌓인 문자가 존재하면 0을 출력하고 stack이 비어있는 상태일 때 1을 출력한다.

 

def solution(s):
    stack = []
    for i in s:
        if len(stack) == 0:
            stack.append(i)
        elif i == stack[-1]:
            stack.pop()
        else:
            stack.append(i)
 
    if len(stack) == 0:
        answer = 1
    else:
        answer = 0
 
    return answer