백준 5430: AC

문제 요약

선영이는 주말에 할 일이 없어서 새로운 언어 AC를 만들었다. AC는 정수 배열에 연산을 하기 위해 만든 언어이다. 이 언어에는 두 가지 함수 R(뒤집기)과 D(버리기)가 있다.

함수 R은 배열에 있는 수의 순서를 뒤집는 함수이고, D는 첫 번째 수를 버리는 함수이다. 배열이 비어있는데 D를 사용한 경우에는 에러가 발생한다.

함수는 조합해서 한 번에 사용할 수 있다.

배열의 초기값과 수행할 함수가 주어졌을 때, 최종 결과를 구하는 프로그램을 작성하시오.

문제 분류

• 구현
• 자료 구조
• 문자열
• 파싱
• 덱

문제 풀이

입력받는 R에 따라 pop할 위치를 앞으로 할지 뒤로 할지 결정하면 된다. 즉, bool타입의 rvs변수를 선언하고 false/true에 따라 D에 pop_front()/pop_back()을 수행하고, 출력 시에는 begin()/rbegin()부터 end()/rend()까지 순회하며 출력하면 된다.

문자열 파싱하는 부분에서 꽤 많이 틀렸다. 다만 문자열을 제대로 스플릿해서 일단 데크에 잘 넣었다면, 나머지는 어렵지 않다.

풀이 코드

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <queue>
#include <string>

using namespace std;

int main()
{
	int t, n, len;
	bool isError, rvs;
	string p, in, sum;
	deque<int> dq;

	scanf("%d", &t);
	while (t--) {
		isError = false;
		rvs = false;
		sum = "";
		cin >> p;
		scanf("%d", &n);
		cin >> in;

		in.erase(0, 1);
		in.erase(in.length() - 1, 1);
		len = in.length();
		for (int i = 0; i < len; i++) {
			if (in[i] == ',') {
				dq.push_back(stoi(sum));
				sum = "";
			}
			else sum += in[i];
		}
		if (n >= 1) {
			dq.push_back(stoi(sum));
		}
		len = p.length();
		for (int i = 0; i < len; i++) {
			if (p[i] == 'R')
				rvs = !rvs;
			else if (p[i] == 'D') {
				if (!dq.empty()) {
					if (rvs) dq.pop_back();
					else dq.pop_front();
				}
				else {
					isError = true;
					break;
				}
			}
		}
		if (isError)
			printf("error\n");
		else {
			printf("[");
			if (rvs) {
				for (auto it = dq.rbegin(); it != dq.rend(); it++) {
					printf("%d", *it);
					if (it != dq.rend() - 1) printf(",");
				}
			}
			else {
				for (auto it = dq.begin(); it != dq.end(); it++) {
					printf("%d", *it);
					if (it != dq.end() - 1) printf(",");
				}
			}
			printf("]\n");
		}
		dq = deque<int>();
	}
	return 0;
}

후일담

std::reverse() 및 string.substring(), string.find() 등을 사용했을 경우 시간초과가 발생했다.