단어 뒤집기 2 - 17413[S3]

설명

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

17413번: 단어 뒤집기 2

문제

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

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

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

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

입력

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

출력

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

풀이

단순 구현 및 스택 사용.

자세한 내용은 코드를 참조할 것

 

스택을 사용하는 풀이 방법은 다음과 같다. (사실 구현이랑 비슷하다.)

1. 문자열을 순회.
2. 태그를 만나지 않았을 때(즉, 일반적인 문자나 단어일 때), 스택에 문자를 하나씩 push한다.
3. 공백을 만나면 스택에 있는 문자들을 pop하여 뒤집은 단어를 결과 문자열에 추가한다. 공백 또한 결과 문자열에 추가한다.
4. <를 만나면 이전에 스택에 남아 있는 문자들을 pop하여 뒤집어 결과 문자열에 추가하고, <부터 >까지 태그 안의 문자열을 직접 결과 문자열에 추가한다. 태그 안에서는 문자들을 스택에 push하지 않는다.
5. 문자열 순회가 끝난 후 스택에 남아 있는 문자들을 pop하여 결과 문자열에 추가한다.

풀이 1 : 구현

이 방법은 상대적으로 난잡했다. 기회가 된다면 스택을 사용해보는 것도 좋을 것 같다.

 

class Main {
    static String[] raw;
    static List<String> words = new ArrayList<>();

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        raw = br.readLine().split(""); //입력 스플릿

        StringBuilder tempStr = new StringBuilder(); //<> 태그 단어가 아닌 일반 단어 보관
        for (int i = 0; i < raw.length; i++) {
            if (raw[i].equals("<")) {
                getReverseStr(tempStr); //저장된 문자열 뒤집어서 저장
                i = getTags(i);
                tempStr = new StringBuilder(); //초기화
            }
            else tempStr.append(raw[i]);
        }
        if (!tempStr.isEmpty()) getReverseStr(tempStr); //저장된 문자열 뒤집어서 저장

        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        for (String now : words) bw.write(now);
        bw.close();
    }

    //단어에 ' ' 공백 포함되어 있을 수 있음
    public static void getReverseStr(StringBuilder stringBuilder) {
        if (stringBuilder.length() == 0) return;
        String[] str = stringBuilder.toString().split(" "); //공백 스플릿
        if(str.length == 1) { //공백 없는 단일 문자일 경우
            words.add(reverse(str[0]));
            return;
        }

        //아닐 경우 중간에 공백 넣어줘야 함
        for(int i = 0; i < str.length; i++) {
            words.add(reverse(str[i]));         //단어 변환
            words.add(new String(" "));  //공백
        }
        words.remove(words.size() - 1);   //마지막 공백 제거
    }

    //글자 뒤집기
    public static String reverse(String input) {
        StringBuilder stringBuilder = new StringBuilder();
        for (int i = input.length() - 1; i >= 0; i--) stringBuilder.append(input.charAt(i));
        return stringBuilder.toString();
    }

    public static int getTags(int idx) {
        StringBuilder sb = new StringBuilder();
        while (true) {
            sb.append(raw[idx]);
            if(raw[idx].equals(">")) {
                words.add(sb.toString());
                return idx;
            }
            idx++;
        }
    }
}

 

풀이 2 : 스택 사용

public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String s = br.readLine();
        StringBuilder result = new StringBuilder();
        Stack<Character> stack = new Stack<>();
        boolean inTag = false;

        for (int i = 0; i < s.length(); i++) {
            char c = s.charAt(i);

            if (c == '<') {
                // 태그 시작 전 스택에 있는 문자 뒤집어서 결과에 추가
                while (!stack.isEmpty()) {
                    result.append(stack.pop());
                }
                inTag = true;
                result.append(c);
            } else if (c == '>') {
                inTag = false;
                result.append(c);
            } else if (inTag) {
                // 태그 안의 문자는 그대로 결과에 추가
                result.append(c);
            } else {
                if (c == ' ') {
                    // 공백 만남: 스택에 있는 문자 뒤집어서 결과에 추가, 공백 추가
                    while (!stack.isEmpty()) {
                        result.append(stack.pop());
                    }
                    result.append(c);
                } else {
                    // 단어의 문자는 스택에 push
                    stack.push(c);
                }
            }
        }

        // 마지막 단어 처리
        while (!stack.isEmpty()) {
            result.append(stack.pop());
        }

        System.out.println(result.toString());
    }
}