선영이는 주말에 할 일이 없어서 새로운 언어 AC를 만들었다. AC는 정수 배열에 연산을 하기 위해 만든 언어이다. 이 언어에는 두 가지 함수 R(뒤집기)과 D(버리기)가 있다.
함수 R은 배열에 있는 수의 순서를 뒤집는 함수이고, D는 첫 번째 수를 버리는 함수이다. 배열이 비어있는데 D를 사용한 경우에는 에러가 발생한다.
함수는 조합해서 한 번에 사용할 수 있다. 예를 들어, "AB"는 A를 수행한 다음에 바로 이어서 B를 수행하는 함수이다. 예를 들어, "RDD"는 배열을 뒤집은 다음 처음 두 수를 버리는 함수이다.
배열의 초기값과 수행할 함수가 주어졌을 때, 최종 결과를 구하는 프로그램을 작성하시오.
첫째 줄에 테스트 케이스의 개수 T가 주어진다. T는 최대 100이다.
각 테스트 케이스의 첫째 줄에는 수행할 함수 p가 주어진다. p의 길이는 1보다 크거나 같고, 100,000보다 작거나 같다.
다음 줄에는 배열에 들어있는 수의 개수 n이 주어진다. (0 ≤ n ≤ 100,000)
다음 줄에는 [x1,...,xn]과 같은 형태로 배열에 들어있는 정수가 주어진다. (1 ≤ xi ≤ 100)
전체 테스트 케이스에 주어지는 p의 길이의 합과 n의 합은 70만을 넘지 않는다.
각 테스트 케이스에 대해서, 입력으로 주어진 정수 배열에 함수를 수행한 결과를 출력한다. 만약, 에러가 발생한 경우에는 error를 출력한다.
4
RDD
4
[1,2,3,4]
DD
1
[42]
RRD
6
[1,1,2,3,5,8]
D
0
[]
[2,1]
error
[1,2,3,5,8]
error
배열을 받을 때는 우선 문자열로 다 받은 후에, 쉼표를 기준으로 쪼개어 덱에 넣어주었다. 2중 while문을 사용하여, 안쪽 while문은 쉼표를 기준으로 끊기도록, 바깥쪽 while문은 ']'를 발견했을 때 끊기도록 하였다.
<- 이렇게 할 경우에는 배열에 들어있는 수의 개수가 0일 때 (즉 [ ]을 입력받았을 때) 쉼표를 발견하지 못해서 0을 덱에 넣어버린다. 따라서 0은 따로 예외처리 하였다.
R은 배열을 뒤집는 함수이고, D는 첫번째 수를 버리는 함수이다.
엥 근데 첫번째 수만 버리면 되는데 왜 큐가 아니라 덱을 사용하지?!
아하!!! R을 입력받았을 때 실제로 배열을 뒤집는 것이 아니라 뒤집어진 척을 하면 되는구나!!!
R이 들어올 때마다 int is_reversed = 0을 0에서 1로, 1에서 0으로 바꾼다.
is_reversed가 0이라면, 뒤집어지지 않은 상태이므로, D가 들어왔을 때 맨 앞 수를 버린다.(deque.pop_front( ))
또한, 뒤집어지지 않은 상태일 때는, 결과를 출력할 때 앞에서부터 출력한다. (deque.front( )와 deque.pop_front( )) 반복
is_reversed가 1이라면, 뒤집어진 상태이므로, D가 들어왔을 때 맨 뒤의 수를 버린다. (deque.pop_back( ))
뒤집어진 상태일 때는 결과를 뒤에서부터 출력하면 되겠지?!
error의 경우에는, deque.empty( )를 이용하였다. D가 들어왔을 때 덱이 비어있다면, error를 출력하고 함수를 바로 종료한다.
#include <iostream>
#include <string>
#include <deque>
using namespace std;
string functions;
string input_arr;
int arr_length;
deque<int> int_deque;
void set_io(void);
void input_data(void);
void ac(void);
void prt_result(int is_reversed);
int main(void)
{
set_io();
int testcase;
cin >> testcase;
while (testcase-- > 0)
{
input_data();
ac();
}
return (0);
}
void set_io(void)
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
return ;
}
void input_data(void)
{
cin >> functions;
cin >> arr_length;
cin >> input_arr;
if (arr_length == 0)
return ;
int i = 0;
while (input_arr[i] != ']')
{
int now_int = 0;
while (input_arr[++i] >= '0' && input_arr[i] <= '9')
{
now_int *= 10;
now_int += input_arr[i] - '0';
}
int_deque.push_back(now_int);
}
return ;
}
void ac(void)
{
int function_length = functions.length();
int i = -1;
int is_reversed = 0;
while (++i < function_length)
{
if (functions[i] == 'R')
is_reversed = is_reversed ? 0 : 1;
else
{
if (int_deque.empty())
{
cout << "error" << '\n';
return ;
}
is_reversed ? int_deque.pop_back() : int_deque.pop_front();
arr_length--;
}
}
prt_result(is_reversed);
return ;
}
void prt_result(int is_reversed)
{
cout << '[';
if (is_reversed == 0)
{
while (!int_deque.empty())
{
cout << int_deque.front();
if (arr_length > 1)
{
cout << ',';
arr_length--;
}
int_deque.pop_front();
}
}
else
{
while (!int_deque.empty())
{
cout << int_deque.back();
if (arr_length > 1)
{
cout << ',';
arr_length--;
}
int_deque.pop_back();
}
}
cout << ']' << '\n';
return ;
}
성공!!^-^