AC - 5430[G5]

문제

https://www.acmicpc.net/problem/5430

5430번: AC

 

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

 

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

 

함수는 조합해서 한 번에 사용할 수 있다. 예를 들어, "AB"는 A를 수행한 다음에 바로 이어서 B를 수행하는 함수이다. 예를 들어, "RDD"는 배열을 뒤집은 다음 처음 두 수를 버리는 함수이다.

 

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

 

풀이

ArrayDeque 사용 + 문자열 사용 문제.

BufferedWriter에서 null이 출력되는 오류를 제외한다면 크게 어려운 것은 없었다.

reverse boolean을 통해 조건문을 걸어 PollLast와 poll을 사용하였다.

 

코드

 

import java.io.*;
import java.util.*;

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

class Main {
	static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
	static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
	static int T;
	public static void main(String[] args) throws IOException {
		T = Integer.parseInt(br.readLine());
		
		while(T-- > 0) {
			StringBuilder command = new StringBuilder(br.readLine());
			int num = Integer.parseInt(br.readLine());
			StringBuilder arr = new StringBuilder(br.readLine());
			solve(command, arr);
		}
		
		bw.close();
	}
	
	static void solve(StringBuilder command, StringBuilder arr) throws IOException {
		boolean reverse = false;
		ArrayDeque<Integer> nowQueue = new ArrayDeque<>();
		int nowNum = 0; //현재 배열 숫자
		
		//1. Array 파싱
		for(int i = 0; i < arr.length(); i++) {
			char now = arr.charAt(i);
			if(now < 48 || now > 57) {
				if(nowNum != 0) { //저장숫자가 있다면 저장
					nowQueue.add(nowNum);
					nowNum = 0;
				}
				continue; //괄호, ',' 배제 및 큐 추가
			}		
			nowNum *= 10; //자리수 증가
			nowNum += (now - '0'); //현재 숫자 추가
		}

		
		//2. 명령어 수행
		for(int i = 0; i < command.length(); i++) {
			char now = command.charAt(i);
			if(now == 'R') reverse = !reverse; //반전
			else { //D의 경우 
				//배열이 비었는데 D일시 에러
				if(nowQueue.isEmpty()) {
					bw.write("error\n");
					return;
				}
				
				if(reverse) nowQueue.pollLast(); //마지막 꺼내기
				else nowQueue.poll();
			}
		}
		
		//3. 출력
		bw.write("[");
		if(!nowQueue.isEmpty()) {
			if(reverse) {
				bw.write("" + nowQueue.pollLast());
				while(!nowQueue.isEmpty()) bw.write("," + nowQueue.pollLast());
			} else {
				bw.write("" + nowQueue.poll());
				while(!nowQueue.isEmpty()) bw.write("," + nowQueue.poll());			
			}
		}
		bw.write("]\n");
	}
	

}