[백준] 3986/좋은 단어

문제

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

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

입력

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

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

출력

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

풀이

스택을 이용하여 풀었다. 문자를 차례차례 스택에 넣었다. 스택이 비어있거나 스택의 가장 최근에 들어간 문자가 현재 문자와 같지 않으면 스택에 문자를 넣어주었고, 가장 최근에 들어간문자와 현재문자가 같으면 스택에서 빼주었다. 이걸 단어의 길이만큼 반복해 주었다. 반복이 끝나고 스택이 비어있으면 좋은단어라고 판단하였다.

여기서 내가 다른사람의 코드와 비교해서 깨달은 것은 문제를 풀고 거기서 끝낼 것이 아니라 다시 재정리하는 시간이 필요하다는 것이었다. '클린코드'에서도 많은 중첩이 있는 코드는 좋은코드가 아니라고 했다. 조건과 반복되는 부분을 정리리하는 버릇이 필요하다.

고치기 전 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n;
        String word;

        n = Integer.parseInt(br.readLine());

        int cnt=0;
        for(int i = 0; i < n ;i++){
            word = br.readLine();
            if(isRightWord(word)) {
                cnt++;
            }
        }
        System.out.println(cnt);
    }

    private static boolean isRightWord(String word){
        Stack<String> stack = new Stack<>();

        for(int i = 0 ; i < word.length() ;i++){
            if(stack.isEmpty()){
                stack.push(String.valueOf(word.charAt(i)));
            } else{
                String top = stack.peek();
                if(top.equals(String.valueOf(word.charAt(i)))){
                    stack.pop();
                } else{
                    stack.push(String.valueOf(word.charAt(i)));
                }
            }
        }
        return stack.isEmpty()? true : false;
    }
}

고친 후 코드(고친 후의 코드 속도가 확실 히 빨라졌다. 아마도 조건문 감소, String 비교연산이 사라져서 인 듯 하다.)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n;
        String word;

        n = Integer.parseInt(br.readLine());

        int cnt=0;
        for(int i = 0; i < n ;i++){
            word = br.readLine();
            if(isRightWord(word)) {
                cnt++;
            }
        }
        System.out.println(cnt);
    }

    private static boolean isRightWord(String word){
        Stack<Character> stack = new Stack<>();

        for(int i = 0 ; i < word.length() ;i++){
            char curChar = word.charAt(i);
            if(stack.isEmpty() || curChar != stack.peek()){
                stack.push(curChar);
            } else{
                stack.pop();

            }
        }
        return stack.isEmpty()? true : false;
    }
}