백준 17413번 단어뒤집기2 문제로 알아보는 논리 연산의 중요성

문제

문자열 S가 주어졌을 때, 이 문자열에서 단어만 뒤집으려고 한다.

먼저, 문자열 S는 아래와과 같은 규칙을 지킨다.

1. 알파벳 소문자('a'-'z'), 숫자('0'-'9'), 공백(' '), 특수 문자('<', '>')로만 이루어져 있다.
2. 문자열의 시작과 끝은 공백이 아니다.
3. '<'와 '>'가 문자열에 있는 경우 번갈아가면서 등장하며, '<'이 먼저 등장한다. 또, 두 문자의 개수는 같다.

태그는 '<'로 시작해서 '>'로 끝나는 길이가 3 이상인 부분 문자열이고, '<'와 '>' 사이에는 알파벳 소문자와 공백만 있다. 단어는 알파벳 소문자와 숫자로 이루어진 부분 문자열이고, 연속하는 두 단어는 공백 하나로 구분한다. 태그는 단어가 아니며, 태그와 단어 사이에는 공백이 없다.

입력

첫째 줄에 문자열 S가 주어진다. S의 길이는 100,000 이하이다.

출력

첫째 줄에 문자열 S의 단어를 뒤집어서 출력한다.

예제 입력 1 복사

baekjoon online judge

예제 출력 1 복사

noojkeab enilno egduj

예제 입력 2 복사

<open>tag<close>

예제 출력 2 복사

<open>gat<close>

예제 입력 3 복사

<ab cd>ef gh<ij kl>

예제 출력 3 복사

<ab cd>fe hg<ij kl>

예제 입력 4 복사

one1 two2 three3 4fourr 5five 6six

예제 출력 4 복사

1eno 2owt 3eerht rruof4 evif5 xis6

예제 입력 5 복사

<int><max>2147483647<long long><max>9223372036854775807

예제 출력 5 복사

<int><max>7463847412<long long><max>7085774586302733229

예제 입력 6 복사

<problem>17413<is hardest>problem ever<end>

예제 출력 6 복사

<problem>31471<is hardest>melborp reve<end>

예제 입력 7 복사

< space >space space space< spa c e>

예제 출력 7 복사

< space >ecaps ecaps ecaps< spa c e>

---

해설

 

import java.io.*;
import java.util.Stack;

public class Main {
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter wr = new BufferedWriter(new OutputStreamWriter(System.out));
        Stack<Character> stack = new Stack<>();
        char[] arr = (br.readLine() + "\n").toCharArray();
        int N = arr.length;
        boolean tag = false;
        for(int i = 0; i < N; ++i){
            char c = arr[i];
            if(c == '<') tag = true;

            // 태그가 되버리거나, 혹은 문장이 끝나거나 혹은 공백을 만났을 때이다.
            if(tag || c == ' ' || c == '\n'){
                while(!stack.isEmpty()) wr.write(stack.pop());
                if(c != '\n') wr.write(c);
            } else {
               stack.push(c);
            }
            if(c == '>') tag = false;
        }
        wr.flush();
    }
}

 

이 문제를 처음 풀 때는 if조건을 무작정 늘려서 작성했다. 하지만 그렇게하니 너무 어지러웠고, 이 소스코드가 정확한 지 나 스스로에게 자신이 없었다. 그래서 한 번 언제 stack에 있는 문자를 pop해야해는 지 분석해보았다. 

태그가 참으로 변하거나, 혹은 문장이 끝나거나 혹은 공백을 만났을 때

위와 같은 경우에 스택에 있는 문자를 모두 pop해줘야만 했다.

 

위 조건을 작성하고 나니, 코드를 어떻게 작성해야할 지 감이 오지 않는가?

위 조건에 해당하는 경우엔 한 번 스택을 비워주면, 이후엔 반복문이 실행되지 않으니, 시간복잡도에도 크게 영향을 끼치지 않는다.

위의 조건식의 모든 조건이 거짓일 경우엔, 스택에 push해주면 문제가 해결된다.

느낀 점

• 조건이 복잡할 땐, 조건을 나열해서 작성해보고 이를 줄일 수 있는 부분이 있는 지를 분석해보는 게 중요하다!!