zz12345의 코딩 공부

www.acmicpc.net/problem/17413

 

17413번: 단어 뒤집기 2

문자열 S가 주어졌을 때, 이 문자열에서 단어만 뒤집으려고 한다. 먼저, 문자열 S는 아래와과 같은 규칙을 지킨다. 알파벳 소문자('a'-'z'), 숫자('0'-'9'), 공백(' '), 특수 문자('<', '>')로만 이루어져

www.acmicpc.net

문제

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

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

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

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

 

코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#include <iostream>
#include <stack>
#include <string>
using namespace std;
void stackprint(stack<char> &stack)
{
    while(!stack.empty())
            {
                cout<<stack.top();
                stack.pop();
            }
}
int main()
{
    stack<char> stack;
    string str;
    getline(cin,str);
    for(int i=0;i<str.size();i++)
    {
        if(str[i]=='<')
        {
            stackprint(stack);
            while(1)
            {
               cout<<str[i];
                if(str[i]=='>')
                    break;
                i++;
            }
        }
        else if(str[i]==' ')
        {
           stackprint(stack);
            cout<<" ";
        }
        else
        {
            stack.push(str[i]);
        }
    }
    stackprint(stack);
}
cs

풀이

태그는( <와 >사이의 문자열) 그대로 출력하고 태그가 아닌 문자열에서 공백을 기준으로 단어를 구분하여 그 단어들의 순서를 뒤집어서 출력해 주는 문제이다.

 

예시 입력 : <ABCD>AP AD JINX<LLOL>

예시 출력 : <ABCD>PA DA XNIJ<LLOL>

 

우선 단어를 뒤집을 땐 스택을 사용한다 단어를 순서대로 스택에 넣어주고 빼주면 순서가 뒤집히기 때문이다.(후입선출)

 

태그의 시작인 ' < '가 등장할 수 있는 시점은 단어의 끝 뒤나, 입력 문자열의 맨 처음이다. 따라서 입력 문자열을 순서대로 검사하다가 ' < '를 만났다면 스택에 있을지도 모르는 단어를 출력해 준다. (20~22행) 그 후에 태그의 내용을 순서대로 출력해 준다 (23~29행) 이를 마치면 입력 문자열을 순서대로 검사하는 반복문은 태그의 바로 뒤를 검사한다.(28행의 i++에 의해서)

 

공백이 나올 수 있는 경우는 단어의 끝이나 태그 안에서 나올 수 있는데. 태그의 안에서 나오는 경우는 위에서 그대로 출력해 주기 때문에 신경 쓰지 않는다. 따라서 공백이라면 무조건 단어의 끝이라고 볼 수 있다. (31번행 else if 가 true일때)그때에는 스택의 모든 내용을 출력해 준다.

 

태그의 시작인 ' < '나 공백 '  ' 이 아니라면 단어의 일부이고 스택에 push 해준다.

 

모든 문자열을 검사했다면(반복문이 끝났다면) 입력 문자열이 단어로 끝났을 경우를 생각해서 스택을 모두 출력한다. 스택이 빌 때까지 출력하는 동작은 stackprint 함수를 만들어서 사용했다. 스택의 empty() 함수가 FALSE를 반환한다면 계속 스택의 top() 을 출력하고 pop() 하는 간단한 함수이다. 문자열에 공백을 포함하여서 입력받기 때문에 getline 함수를 사용했다.

공유하기

facebook twitter kakaoTalk kakaostory naver band