백준 17413번 단어뒤집기 2 java

문제

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

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

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

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

입력

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

출력

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

 

 

 

앞서 풀어보았던 단어뒤집기1 문제에서 조금만 추가하면 되는 문제다.

문자열 '<' 일 때, 경우의 수는 두가지이다.

1. <hello> 처럼 문장의 시작 일 때

2. hello<hihi> 처럼 문장의 중간일때

 

1번과 2번 모두 '<' 앞에있는 문자를 거꾸로 출력해줘야 한다. 1번의 경우도 그냥 쉽게 생각해서, 공백을 거꾸로 출력한다고 생각하자. (공백은 거꾸로 출력해도 공백)

 

'<' 와 '>' 사이를  boolean 변수로, 위 사항만 고려한다면 쉽게 풀어나갈 수 있는 문제다.

 

 

 

import java.util.Scanner;
import java.util.Stack;

public class Main {

	static void print(Stack st) {
		while (!st.empty()) {
			System.out.print(st.peek());
			st.pop();
		}
	}

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		Stack st = new Stack();

		String s = sc.nextLine();
		boolean check = false;

		for (int i = 0; i < s.length(); i++) {
			if (s.charAt(i) == '<') {
				print(st); // <를 만나면 여태껏 저장해둔 문자열 거꾸로 출력
				check = true;
				System.out.print(s.charAt(i));
			} else if (s.charAt(i) == '>') {
				check = false;
				System.out.print(s.charAt(i));
			} else if (check) {
				System.out.print(s.charAt(i));
			}

			else {
				if (s.charAt(i) == ' ') {
					print(st);
					System.out.print(s.charAt(i));
				}

				else {
					st.push(s.charAt(i));
				}
			}

		}

		print(st);

	}

}