[Programmers] 짝지어 제거하기

문제

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

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

b aa baa → bb aa → aa →

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

제한사항

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

---

풀이

풀이1 (시간초과)

첫번째 풀이는 시간초과로 통과하지 못한 코드이다.

문자열의 처음부터 끝까지 돌면서 위의 글자와 같으면

replace로 문자열 내의 모든 짝지어진 글자를 없애고 다시 처음부터 탐색하는 방법이다.

최대 시간인 경우를 가정해 보자면 'abcdefghijklmnopqrstuvwxyz ... zaa' 이렇게 맨 끝에 반복되는 수가 있을 경우,

i가 999,998까지 갔다가 다시 0으로 돌아와서 다시 999,998이 되어야 종료될 수 있다.

def solution(s):
    i = 0
    while True:
        
        if s[i] == s[i+1]:
            s = s.replace(s[i]*2, "")
            i = 0
        else:
            i += 1
        
        if i == len(s)-1:
            return 0
        elif len(s) == 0:
            return 1

 

풀이2

풀이를 전혀 모르겠어서 스택이라는 힌트를 얻고 다시 풀어봤다.

1. 빈 스택이면 일단 비교하는 값을 넣는다.

2. 스택이 채워져 있다면 스택에 있는 마지막 값과 비교한다.

2-1. 같다면 스택을 비운다.

2-2. 다르다면 스택에 추가한다.

위 과정을 반복하면 최대 1,000,000번만 하면 가능여부를 확인할 수 있다.

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