백준 3986번 Pseudo code

문제 링크

https://www.acmicpc.net/problem/3986

3986번: 좋은 단어

 

시간 제한 / 메모리 제한

1 초

256 MB

 

문제

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

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

 

입력

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

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

출력

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

예제 입력

3
ABAB
AABB
ABBA

예제 출력

2

 

문제 풀이를 위해 생각한 것

1. 선이 교차하지 않는다는 것은, 먼저 등장한 글자가 그 다음에 등장한 글자보다 나중에 짝을 찾는 것.
2. 대표적인 스택 문제인 괄호 검사와 유사하다.

 

사용한 자료구조

스택.

이유는?

괄호 검사 하듯이 한 단어 쌍의 등장과 재등장을 스택을 통해 검사할 수 있음.

 

풀이 코드

n = 첫 입력값을 정수로 변환
stk = 스택 자료구조 선언
cnt = 0


N개의 입력값에 대하여:
	입력값의 각 글자 ch에 대하여:
		if stk이 비어있으면:
			stk.push(ch)
		elif stk에 원소가 있으면:
			if ch가 stk의 최상단 값과 같으면:
				stk.pop()
			elif ch가 stk의 최상단 값과 다르면:
				stk.push()

	if 검사가 끝나고 stk에 원소가 없으면:
		cnt += 1

print cnt

 

시간 복잡도 분석

O(N). 

 

스택의 삽입, 삭제는 O(1)이다.

N개의 단어의 길이만큼 O(1) 의 스택 삽입 삭제 연산을 수행해야 한다.

시간 제한은 1초(대략 1억 번 연산 가능)이지만 모든 단어의 길이 합이 1,000,000을 넘지 않기 때문에 O(1) * 1,000,000 으로시간 제한 내에 문제를 해결할 수 있다.

 

문제에서 중요한 부분

원소가 없을 때와 있을 때의 연산을 구분해주어야 한다.

파이썬으로 푼다면 sys.stdin.readline으로 입력받는 시간을 줄이자.