BOJ_G5_5430

📁 [G5_5430] AC

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.StringTokenizer;


public class Main {
	static ArrayDeque<Integer> deque;
	static String str, error;
	static int N;
	static StringBuilder sb;
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st;
		sb = new StringBuilder();

		// 테케
		int TC = Integer.parseInt(br.readLine());

		for (int t = 0; t < TC; t++) {
			// 함수 입력
			str = br.readLine();

			// 수의 개수 입력
			N = Integer.parseInt(br.readLine());

			// error 저장
			error = "";

			// ArrayList ( 수 저장 리스트 )
			deque = new ArrayDeque<Integer>();
			st = new StringTokenizer(br.readLine(), "[,]");
			for (int i = 0; i < N; i++) {
				deque.add(Integer.parseInt(st.nextToken()));
			}

			go();
		}
		System.out.println(sb);
	}
	static void go() {
		// 앞인지 뒤인지 ( 앞 true, 뒤 false )
		boolean check = true;

		for(int i=0; i<str.length(); i++) {
			// R 함수인 경우 방향 바꾸기
			if(str.charAt(i) == 'R') {
				check = !check;
			}
			// D 함수 인 경우
			else if(str.charAt(i) == 'D') {
				// 배열이 정방향인 경우
				if(check) {
					// 크기가 없는 경우
					if(deque.size()<1) {
						error = "error";
					}else {
						deque.removeFirst();
					}
				}
				// 배열이 반대인 경우
				else {
					// 크기가 없는 경우
					if(deque.size()<1) {
						error = "error";
					}else {
						deque.removeLast();
					}
				}
			}
		}
		print(check);
	}

	static void print(boolean check) {
		// error 인 경우
		if(error.equals("error")) {
			sb.append("error").append("\n");
		}
		// 아닌 경우
		else {
			sb.append("[");
			// 정방향인 경우 그대로 출력
			if(check) {
				int a = deque.size();
				for(int i=0; i<a;i++) {
					sb.append(deque.poll());
					// 쉼표
					if(i != a-1) {
						sb.append(",");
					}
				}
			}
			// 아닌 경우 반대로 출력
			else {
				int a = deque.size();
				for(int i=0; i<a;i++) {
					sb.append(deque.pollLast());
					// 쉼표
					if(i != a-1) {
						sb.append(",");
					}
				}
			}
			sb.append("]").append("\n");
		}
	}
}

🤔 나의 생각

문제는 쉽게 보였지만 .. 역시 시간초과 ..
가장 중요한 부분은 R 함수를 어떻게 처리하냐 인 것 같다.
그냥 이중 for문으로 처리하니 p와 n 값이 커서 시간초과가 뜬다.
그래서 생각해야 하는 것은 deque를 사용하는 것이다.
deque를 통해 R 함수가 호출될 때는 boolean 값을 이용해 앞인지 뒤인지만 체크해주고 D 함수가 호출될 때는 firstremove() 과 lastremove()을 통해 값을 지워주고
다 출력할 때는 boolean 값에 따라 firstpoll()과 lastpoll()을 이용하면 된다.