[백준] 좋은 단어 (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

---

주어지는 문자열이 좋은 단어인지 판별하는 문제다.

좋은 단어란 같은 글자끼리 이었을 때 이은 선이 교차하지 않는 단어이다.

예를들어 ABAB는 A끼리 잇는 선과 B끼리 잇는 선이 교차하지만, 

AABB나 ABBA는 교차하지 않는다.

 

해결방법은 스택이다. 스택에 글자를 하나씩 쌓으면서 스택의 탑과 같은 글자가 들어오면 pop하고 아니면 push한다.

주의할 점은 스택이 비어있는 경우를 항상 신경써야 한다는 것이다. 

스택 문제에서 오류 80% 이상이 empty인 경우를 고려하지 않아서 생기는 오류였다.

따라서 항상 스택이 비어있는지 체크해서, 비어있다면 그 글자를 그대로 push 해주자.

 

스택에 단어를 넣고 빼는 과정을 마무리했는데, 스택의 크기가 0이 아니라면 잘못된 단어다.

좋은 단어라면 같은 글자를 넣고 빼며 크기가 0이 되어야 한다.

 

 

코드 원본 : https://github.com/chosh95/STUDY/blob/bebc3c7f253fe8f425ef7a8a340730d6f9cc35bc/BaekJoon/2020/%EC%A2%8B%EC%9D%80%20%EB%8B%A8%EC%96%B4%20(3986%EB%B2%88).cpp

chosh95/STUDY

 

C++ 코드

#include <iostream>
#include <string>
#include <stack>
#include <algorithm>
using namespace std;
int N;
int res;

int main()
{
	cin >> N;
	while (N--) {
		string str;
		cin >> str;
		stack<char> st;
		for (int i = 0; i < str.size(); i++) {
			if (st.empty()) {
				st.push(str[i]);
				continue;
			}
			if (st.top() == str[i]) {
				st.pop();
			}
			else {
				st.push(str[i]);
			}
		}
		if (st.empty())
			res++;
	}
	cout << res;
}