[201] 17413번 단어 뒤집기2

17413번 단어 뒤집기2

문제

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

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

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

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

입력

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

출력

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

예제 입력 1 복사

baekjoon online judge

예제 출력 1 복사

noojkeab enilno egduj

예제 입력 2 복사

<open>tag<close>

예제 출력 2 복사

<open>gat<close>

예제 입력 3 복사

<ab cd>ef gh<ij kl>

예제 출력 3 복사

<ab cd>fe hg<ij kl>

예제 입력 4 복사

one1 two2 three3 4fourr 5five 6six

예제 출력 4 복사

1eno 2owt 3eerht rruof4 evif5 xis6

예제 입력 5 복사

<int><max>2147483647<long long><max>9223372036854775807

예제 출력 5 복사

<int><max>7463847412<long long><max>7085774586302733229

예제 입력 6 복사

<problem>17413<is hardest>problem ever<end>

예제 출력 6 복사

<problem>31471<is hardest>melborp reve<end>

예제 입력 7 복사

<   space   >space space space<    spa   c e>

예제 출력 7 복사

<   space   >ecaps ecaps ecaps<    spa   c e>

코드

//---- 세팅 ----//
const fs = require('fs');
const stdin = (
  process.platform === 'linux'
    ? fs.readFileSync('/dev/stdin').toString()
    : `\
baekjoon online judge
`
).split('\n');

const input = (() => {
  let line = 0;
  return () => stdin[line++];
})();

//---- 풀이 -----//

const str = input();
const strArr = str.split('');
const res = [];

const isNum = c => {
  if (c.trim() === '') return false;
  return [0, 1, 2, 3, 4, 5, 6, 7, 8, 9].includes(Number(c));
};

const isAlphabet = c => {
  const code = c.charCodeAt();
  if (65 <= code && code <= 90) return true;
  if (97 <= code && code <= 122) return true;
  return false;
};

while (strArr.length > 0) {
  const stack = [];

  if (strArr[0] === '<') {
    while (strArr[0] !== '>') {
      stack.push(strArr.shift());
    }
    stack.push(strArr.shift());
    res.push(...stack);
  } else if (isNum(strArr[0]) || isAlphabet(strArr[0])) {
    while (strArr.length > 0 && strArr[0] !== '<' && strArr[0] !== ' ') {
      stack.push(strArr.shift());
    }
    stack.reverse();
    res.push(...stack);
  } else {
    res.push(strArr.shift());
  }
}

console.log(res.join(''));

풀이

기존에 단어 뒤집기1 문제 같은 경우는 공백기준으로 쪼개고 뒤집기만 하면됬다.

하지만 이번 문제는 조금더 복잡한 경우이므로 스택을 활용하였다.

우선 입력 받은 모든 문자들을 하나의 문자로 전부 쪼개서 배열에 넣어놓는다.

그리고 아래와 같은 3가지 경우로 문자들을 스택에 쌓고 res에 푸쉬한다.

1. '<' 로 시작하는 경우 '>' 까지 unshift하여 stack에 쌓고 stack을 다시 res에 push한다.
2. 문자 혹은 숫자로 시작하는경우 '<' 혹은 ' '(공백) 이전까지 unshift하여 stack에 쌓고 reverse한뒤 res에 푸쉬한다.
3. 위에 케이스 이외의 경우 (예를들어 공백인 경우), unshift하여 곧 바로 res에 push한다.