[Stack] 백준 3986 문제풀이

처음에는 문제가 이해가 안갔는데 그냥 A와 A를 짝짓고 B와 B를 짝지었을 때 겹치는 선이 없고 남는 문자가 없으면 좋은 문자로 세어주는 것이었다. 

이 문제를 푸는 알고리즘은 

1. 스택에 같은 문자가 들어오면 pop

2. 스택에 다른 문자가 들어오면 push 하는 것이다. 

 

파이썬으로 알고리즘을 짜기 시작한지 얼마 되지 않아 리스트가 비어있을 경우에 대한 조건문을 만들때 아래와 같은 코드를 사용하는 것이 len(stack)을 사용하는 것보다 훨씬 나은 코드라는 것도 알게 되었다. 

 

if not stack:

 

그리고 리스트의 마지막 원소에 접근할 때 stack[-1]처럼 사용한다는 것을 알게 되었다.

 

def main():
    n = int(input())
    word_list = []
    stack=[]
    count = 0

    for _ in range(n):
        word_list.append(input())

    for word in word_list:
        for i in range(0,len(word)):
            if not stack:
                stack.append(word[i])
            else:
                if stack[-1]==word[i]:
                    stack.pop()
                else:
                    stack.append(word[i])
        if not stack:
            count+=1
        stack=[]
    print(count)


if __name__ == '__main__':
    main()