17413. 단어 뒤집기 2 (자바, Java)

쉽지만 어려웠던...문제라고 해야하나,

창의력이 모자라서, 풀이 힌트 보고 만들어서 풀었습니다. 코드까지 본건 아니지만, 자력으로는 해결 못한셈...

핵심은 태그 상태 및 공백이 있으면 그대로 입력, 그 외의 경우는 공백이나 <가 나올때까지 스택에 쌓았다가 한번에 pop으로 방출하는 것.

 

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

public class Main {

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String str = br.readLine();
		Stack <Character> stack = new Stack<>();
		StringBuilder sb = new StringBuilder();
		boolean tag = false;
		for (int i=0; i<str.length(); i++) {
			char ch = str.charAt(i);
			if (ch=='<') {
				while (!stack.isEmpty()) {
					sb.append(stack.peek());
					stack.pop();
				}
				tag = true;
				sb.append('<');
			} else if (ch==' ') {
				while (!stack.isEmpty()) {
					sb.append(stack.peek());
					stack.pop();
				}
				sb.append(' ');
			} else if (ch=='>') {
				tag = false;
				sb.append('>');
			} else if (tag) {
				sb.append(ch);
			} else {
				stack.push(ch);
			} 
			
			if (i==str.length()-1) {
				while (!stack.isEmpty()) {
					sb.append(stack.peek());
					stack.pop();
				}
			}
		}
		System.out.println(sb);
	}

}