[백준][JAVA(자바)][스택(stack)] #3986 좋은 단어

<문제>

3986번: 좋은 단어

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

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

[입력]

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

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

[출력]

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

---

<풀이과정>

문제를 이해하는데 오래 걸렸다. 

"만약 선끼리 교차하지 않으면서 각 글자를 정확히 한 개의 다른 위치에 있는 같은 글자와 짝 지을 수 있다면, 그 단어는 '좋은 단어'이다." 이 부분이 너무 어려워서 설명을 해준 블로그를 찾아 설명을 보고 코드는 스스로 짜 보도록 했다.

 

우선, 좋은 단어의 예시를 보자.

좋은 단어 예시

첫 번째, ABAB의 경우 아치형 곡선이 겹치는 것을 볼 수 있다. 이와 같은 경우에는 좋은 단어가 되지 못한다.

하지만 두 번째, 세 번째의 경우 아치형 곡선이 겹치지 않기 때문에 좋은 단어가 될 수 있다.

 

예시를 통해서 알 수 있는 것이 있다. A나 B 둘 중 하나라도 인접해 있는 경우라면 좋은 단어가 된다는 것이다. 

이를 통해 얻은 스택을 이용한 규칙은 다음과 같다.

1. 입력받은 문자열을 통해 스택에 데이터를 추가한다.

2. 스택의 맨 위 데이터와 비교하고자 하는 문자열의 문자가 같다면 삭제한다.

위의 규칙이 성립하면 스택은 비어있을 것이고 규칙이 성립되지 않으면 스택은 비어있지 않을 것이다.

그렇다면, 스택이 비어있다면 좋은 단어 임을 알 수 있다.

---

import java.util.Scanner;
import java.util.Stack;
 
public class Main{
 
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        
        int num = sc.nextInt();
        
        String[] s = new String[num];    //String 배열이 아닌 그냥 String을 사용해도 무방하다.
        Stack<Character> st = new Stack<>();
        
        int good = 0;
        
        for(int i = 0; i < num; i++) {
            s[i] = sc.next();
            for(int j = 0; j < s[i].length(); j++) {
                if(st.size() > 0 && st.peek() == s[i].charAt(j)) {
                    st.pop();
                }else {
                    st.add(s[i].charAt(j));
                }
            }
            if(st.size() == 0) {
                good += 1;
            }
            st.clear();
        }
        System.out.println(good);
    }
}

---

<참고 자료>
티스토리, "백준 3986번 좋은 단어 [Stack] :: 마이구미", https://mygumi.tistory.com/99 [마이구미의 HelloWorld]