[자료구조] stack

Stack : Last In First Out의 형태로 자료를 담는 자료구조

간단하게 파이썬 리스트를 생각하면 된다. 

lst = [1,2,3,4]

# push -> append, pop -> pop
lst.append(5) 
lst.pop() 

# 제거하지 않고 접근만 하려면 -> [-1]
lst[-1]

문제)

짝지어 제거하기

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

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

b aa baa → bb aa → aa →

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

 

풀이)

def solution(s):
    lst = list(s) # lst 에서 result로 한개씩 옮겨주면서 같은 문자끼리 만나면 제거해나갈 것이다. 

    if len(lst) % 2 == 1: # 문자의 길이가 홀수이면 짝지어 제거하기를 할 수 없다. 
        return 0
    
    result = [lst.pop()] # 일단 비었으니 하나를 넣어주자.
    
    
    while lst: # lst에 원소가 남아있는 동안 반복한다. 
        if not result: # 만약 result가 empty이면 하나를 넣어준다. 
            result = [lst.pop()]
        
        if result[-1] == lst[-1]: # 옮기기 전에 비교해서 서로 같으면 짝지어 제거해준다. 
            result.pop()
            lst.pop()
        else: result.append(lst.pop()) # 서로 같지 않으면 옮겨준다. 
    
    
    if not result: # 최종적으로 result가 비었다면 성공, 아니라면 실패   
        return 1
    else:
        return 0

stack 개념을 모를땐 어려웠는데 

stack 사용하여 접근하니 생각보다 쉬웠다.