[Python] 백준 3986번

# 자료구조

3986. 좋은 단어 (silver 4)

이번 계절학기에 심리학 개론을 수강 중인 평석이는 오늘 자정까지 보고서를 제출해야 한다. 보고서 작성이 너무 지루했던 평석이는 노트북에 엎드려서 꾸벅꾸벅 졸다가 제출 마감 1시간 전에 깨고 말았다. 안타깝게도 자는 동안 키보드가 잘못 눌려서 보고서의 모든 글자가 A와 B로 바뀌어 버렸다! 그래서 평석이는 보고서 작성을 때려치우고 보고서에서 '좋은 단어'나 세보기로 마음 먹었다.

평석이는 단어 위로 아치형 곡선을 그어 같은 글자끼리(A는 A끼리, B는 B끼리) 쌍을 짓기로 하였다. 만약 선끼리 교차하지 않으면서 각 글자를 정확히 한 개의 다른 위치에 있는 같은 글자와 짝 지을수 있다면, 그 단어는 '좋은 단어'이다. 평석이가 '좋은 단어' 개수를 세는 것을 도와주자.

 

입력

첫째 줄에 단어의 수 N이 주어진다. (1 ≤ N ≤ 100)

다음 N개 줄에는 A와 B로만 이루어진 단어가 한 줄에 하나씩 주어진다. 단어의 길이는 2와 100,000사이이며, 모든 단어 길이의 합은 1,000,000을 넘지 않는다.

 

출력

첫째 줄에 좋은 단어의 수를 출력한다.

 

예제 입력1

3
ABAB
AABB
ABBA

예제 출력1

2

 

예제 입력2

3
AAA
AA
AB

예제 출력2

1

 

예제 입력3

1
ABBABB

예제 출력3

1

 

 

풀이

단어의 수 n을 먼저 입력받고 저장한다. 그 다음 좋은 단어의 개수에 해당하는 cnt를 0으로 초기화해주었다.

for문을 사용해서 n개만큼 str에 단어를 입력받고, rstrip를 사용해서 단어 사이의 공백을 제거해주었다.

좋은 단어를 판별해줄 배열인 stack을 선언해준다.

for문과 if문을 사용해서 만약 stack에 str이 없는 경우 append를 사용해서 해당 단어를 추가해준다. 만약 이미 stack에 추가된 단어인 경우, 또 다른 if문을 사용해서 str에 있는 글자가 이전에 담긴 글자와 같은 경우, 좋은 단어이기 때문에 stack에서 빼내주었다. 

다음과 같은 과정을 거친 후 스택에 아무런 값도 남아있지 않으면 좋은 단어이기 때문에 cnt를 하나 더해주었고, 마지막 결과값으로 cnt를 반환해주었다.

import sys
input = sys.stdin.readline
 
n = int(input())
cnt = 0   # 좋은 단어 개수
 
for i in range(n):
    str = input().rstrip()  # str에 있는 공백 제거
    stack = []
 
    for i in range(len(str)):
        if stack:   # stack에 이미 추가된 단어인 경우
            if str[i] == stack[-1]:
                stack.pop()
            else:
                stack.append(str[i])
        else:
            stack.append(str[i])   # stack에 없는 단어 추가
 
    if not stack:
        cnt += 1
print(cnt)

실행 결과