알고리즘 기초 - 단어 뒤집기2

2024. 2. 20. 17:38알고리즘

문제(백준 17413)

문자열 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>

📝풀이 과정

태그와 단어를 분리하는 것이 가장 중요하다고 생각했다.

그러기 위해서는 현재 탐색하는 것이 태그인지 단어인지 구분해야한다.

이런 경우에는 플래그 값을 이용하면 된다. isTag라는 플래그 값을 이용한다.

그리고 단어들 간의 분류는 띄워쓰기로 분류한다.

즉, 띄워쓰기가 등장하면 단어 하나가 끝났다는 의미이고, 끝난 단어는 뒤집어야한다.

let input = require('fs').readFileSync(0).toString().trim();

function solution(input) {
  let answer = '';
  let tmp = '';
  let isTag = false;
  for (const x of input) {
    if (x === ' ') {
      if (!isTag) answer += [...tmp].reverse().join('') + ' ';
      else answer += tmp + ' ';
      tmp = '';
    } else if (x === '<') {
      isTag = true;
      answer += [...tmp].reverse().join('') + x;
      tmp = '';
    } else if (x === '>') {
      isTag = false;
      answer += tmp + x;
      tmp = '';
    } else {
      tmp += x;
    }
  }
  answer += [...tmp].reverse().join('');
  return answer;
}

console.log(solution(input));