BOJ - 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

사용할 자료구조

가장 최근에 입력된 알파벳과 같은 알파벳이면 pop 되야하기 때문에 스택을 사용해서 구현하였습니다.

풀이 과정

1) ex)

• ABAB의 경우, A와 A를 잇는 아치형 곡선과 B와 B를 잇는 아치형 곡선이 서로 겹치므로 좋은 단어가 아니다.
• ABBA의 경우, 각 알파벳을 잇는 아치형 곡선이 겹치지 않으므로 좋은 단어이다.

2) 스택의 top이 input과 같을 경우에는 pop을 해줘야합니다.

3) 만약 다를 경우나 스택이 비어있는 경우에는 스택에 push해줍니다.

string str;
        cin>>str;
        for(int j=0;j<str.size();j++){
            if(st.empty()){
                st.push(str[j]);
            }else{
                if(st.top()==str[j]){
                    st.pop();
                }else{
                    st.push(str[j]);
                }
            }
        }

4) 위 과정을 반복했을 때 스택이 비어있으면 "좋은 단어"이고 스택에 무언가가 남아있으면 "좋은 단어"가 아닙니다.

시간 복잡도

스택의 삽입/삭제 연산은 O(1) 입니다.

N개 만큼 삽입/삭제 연산을 수행하기 때문에 시간복잡도는 O(N) 입니다.

#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <cstdio>
#include <stack>
#include <queue>
#include <map>
#include <functional>
using namespace std;

int main(){
    cin.tie(0);
    cout.tie(0);
    ios::sync_with_stdio(false);
    int n;
    int answer = 0;
    cin>>n;
    for(int i=0;i<n;i++){
        stack<char> st;
        string str;
        cin>>str;
        for(int j=0;j<str.size();j++){
            if(st.empty()){
                st.push(str[j]);
            }else{
                if(st.top()==str[j]){
                    st.pop();
                }else{
                    st.push(str[j]);
                }
            }
        }
        if(st.empty()){
            answer++;
        }
    }
    cout<<answer;

    return 0;
}