[BaekJoon 17413번 / JAVA] 단어 뒤집기 2

Question

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

17413번: 단어 뒤집기 2

 

---

💡 Solution

단어를 뒤집는다는 점에서 후입선출의 Stack 자료구조를 이용하여 문제를 해결하였습니다.

 

💻 Code

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

public class BJ17413 {

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		Stack<Character> stack = new Stack<>();  //뒤집을 단어 처리용
		StringBuilder sb = new StringBuilder();
		
		String str = br.readLine();
		boolean flag = false;  //열린 꺽새와 닫힌 꺽새 판별용
		for (int i = 0; i < str.length(); i++) {
			if (str.charAt(i) == '<') {
				while (!stack.isEmpty()) {
					sb.append(stack.pop());
				}
				flag = true;
				
			} else if (str.charAt(i) == '>'){
				sb.append('>');
				flag = false;
				continue;
			}
			
			if (flag) {
				sb.append(str.charAt(i));
			} else if (!flag) {
				if (str.charAt(i) == ' ') {
					while (!stack.isEmpty()) {
						sb.append(stack.pop());
					}
					sb.append(' ');
				} else {
					stack.push(str.charAt(i));
				}
			}
			
			//스택에 남아있는 단어가 있다면
			if (i == str.length() - 1) {
				while (!stack.isEmpty()) {
					sb.append(stack.pop());
				}
			}
			
		}
		System.out.println(sb.toString());
	}

}