[백준] 17413번: 단어 뒤집기 2/ Java 11

• 실패, 정답보고 이해하고 풀이

https://www.acmicpc.net/problem/17413

단어 뒤집기 2

 

시간 제한메모리 제한제출정답맞힌 사람정답 비율

1 초

512 MB

39057

22104

17143

57.005%

문제

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

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

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

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

입력

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

출력

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

 

나의 답안

/*
* 단어 뒤집기
* - <> 태그 안에 있는 단어는 뒤집지 않고 그대로 출력
* - 그 외의 단어들은 스택에 담았다가 pop() 하여 거꾸로 꺼내기*/
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;
import java.util.StringTokenizer;

public class Main{
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        Stack<Character> stack = new Stack<>();

        String str = br.readLine();
        int len = str.length();

        // 태그가 있는지 확인
        boolean tag = false;
        for(int i = 0; i < len; i++){
            if(str.charAt(i) == '<'){
                tag = true;

                while( !stack.isEmpty()){
                    sb.append(stack.pop());
                }

                sb.append(str.charAt(i));

            } else if(str.charAt(i) == '>'){
                tag = false;
                sb.append(str.charAt(i));
            } else if(tag){
                // 태그 안 단어는 그대로 출력
                sb.append(str.charAt(i));
            } else if (!tag) {
                if(str.charAt(i) == ' ') {

                    while( !stack.isEmpty() ) {
                        sb.append(stack.pop());
                    }

                    sb.append(str.charAt(i));
                }
                // 공백이 아닐경우, 공백을 만날 때 까지 stack 에 push
                else {
                    stack.push(str.charAt(i));
                }
            }
        }
        // 마지막 문자열이 남아있는지 체크
        while( !stack.isEmpty() ) {
            sb.append(stack.pop());
        }

        System.out.println(sb);

    }
}