백준 3986번 좋은 단어 파이썬

문제링크

문제

이번 계절학기에 심리학 개론을 수강 중인 평석이는 오늘 자정까지 보고서를 제출해야 한다. 보고서 작성이 너무 지루했던 평석이는 노트북에 엎드려서 꾸벅꾸벅 졸다가 제출 마감 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

제출 답안

import sys
input = sys.stdin.readline

n = int(input())
stack = [input().split()[0] for i in range(n)]
cnt = 0
for word in stack:
    if len(word) % 2 != 0:
        continue
    else:
        stack2 = [word[0]]
        for i in range(1,len(word)):
            if len(stack2) == 0:
                stack2.append(word[i])
                continue
            if word[i] == stack2[-1]:
                stack2.pop()
            else:
                stack2.append(word[i])
    if len(stack2) == 0:
        cnt += 1
print(cnt)

1. 홀수면 그냥 지나가고 짝수부터 비교한다.
2. 우선 stack2에 단어의 0번째를 1부터 시작한다.
3. stack2가 비었으면 단어를 추가하고 stack2[-1]과 지금 들어온 인덱스의 단어가 같으면 둘이 아치형의 곡선이 된다는 뜻이다. 이렇게 비교해가고 둘다아니면 stack2에 추가한다.
4. 최종적으로 스택이 비었으면 모두 아치형이 된다는 뜻이고 남아있다면 안된다는 뜻이다. 0이면 1을 더해준다.

프린트해보기

import sys
input = sys.stdin.readline

n = int(input())
stack = [input().split()[0] for i in range(n)]
cnt = 0
for word in stack:
    if len(word) % 2 != 0:
        continue
    else:
        stack2 = [word[0]]
        for i in range(1,len(word)):
            if len(stack2) == 0:
                print(f'첫번째if문{word[i]},{stack2}')
                stack2.append(word[i])
                continue
            if word[i] == stack2[-1]:
                print(f'두번째if문{word[i]},{stack2[-1]},{stack2}')
                stack2.pop()
            else:
                print(f'마지막{stack2}')
                stack2.append(word[i])
    if len(stack2) == 0:
        cnt += 1
print(cnt)

출력

마지막['B']
마지막['B', 'A']
두번째if문B,B,['B', 'A', 'B']
두번째if문A,A,['B', 'A']
두번째if문B,B,['B']
1
1