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

문제 설명

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

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

b aa baa → bb aa → aa →

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

제한사항

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

---

입출력 예

s result

baabaa	1
cdcd	0

입출력 예 설명

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

 

코드

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

 

풀이

처음 문제를 풀 때 while문 안의 for문을 이용하여 i번째와 i+1번째가 같다면 제거하고, 제거 후 i-1 번째와 i번째를 비교하여 같다면 제거하는 방식으로 풀려고 했다. 그러나 for문 안에 또 for문을 쓰게되게끔 코드가 만들어져서 코드가 많이 복잡해졌다. 

 

스택을 이용하면 이를 해결할 수 있었다. stack을 하나 만들어 stack에 값을 넣어주고 마지막으로 넣은 문자와 해당 순서의 문자가 동일하다면 제거해주고, 같다면 다시 stack에 쌓아주는 방법이다. 마지막까지 비교하여 스택에 값이 존재한다면 0을 반환해주고, 스택에 값이 존재하지 않는다면 1을 반환해 주는 방식으로 문제를 풀 수 있었다.