[백준] 5430 - AC (자바 Java)

[Gold V] AC - 5430

문제 링크

성능 요약

메모리: 103140 KB, 시간: 728 ms

분류

덱(deque), 파싱(parsing), 구현(implementation), 문자열(string), 자료 구조(data_structures)

문제 설명

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

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

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

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

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다. T는 최대 100이다.

각 테스트 케이스의 첫째 줄에는 수행할 함수 p가 주어진다. p의 길이는 1보다 크거나 같고, 100,000보다 작거나 같다.

다음 줄에는 배열에 들어있는 수의 개수 n이 주어진다. (0 ≤ n ≤ 100,000)

다음 줄에는 [x₁,...,x_n]과 같은 형태로 배열에 들어있는 정수가 주어진다. (1 ≤ x_i ≤ 100)

전체 테스트 케이스에 주어지는 p의 길이의 합과 n의 합은 70만을 넘지 않는다.

출력

각 테스트 케이스에 대해서, 입력으로 주어진 정수 배열에 함수를 수행한 결과를 출력한다. 만약, 에러가 발생한 경우에는 error를 출력한다.

---

풀이

 

문자열을 파싱하고, 덱을 사용하는 문제였다.

 

split을 사용해서 String 배열로 만든 후에, deque에 int형으로 바꾸어 하나씩 추가하였다.

 

deque을 사용한 이유는 앞과 뒤에서 모두 remove를 할 수 있기 때문이다.

 

처음에는 first라는 boolean 변수를 true로 두고, R연산이 있을 때마다 first를 바꿨다.

 

first가 true일 경우 D연산에서 removeFirst를, false일 경우 removeLast를 수행한다.

 

연산이 모두 끝난 후 출력할 때도 first에 따라 출력하였다.

 

간과한 부분은 출력할 때 원소마다 val+","를 sb에 추가하고 마지막에는 콤마를 제거하였는데, 배열이 비어있을 경우에 대활호 '['가 제거되는 문제였다.

 

빈 배열일 경우에는 따로 처리해주는 것으로 해결하였다.

 

package BOJ.Gold.g5;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayDeque;
import java.util.Deque;
import java.util.stream.Stream;

public class BOJ_5430_AC {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int T = Integer.parseInt(br.readLine());
        StringBuilder sb = new StringBuilder();
        out:
        while (T-- > 0) {
            String p = br.readLine();
            int n = Integer.parseInt(br.readLine());
            String input = br.readLine();

//            int[] arr = Stream.of(input.substring(1, input.length() - 1).split(",")).mapToInt(Integer::parseInt).toArray();
            String[] arr = input.substring(1, input.length() - 1).split(",");
            Deque<Integer> deque = new ArrayDeque<>();
            for (int i = 0; i < n; i++)
                deque.addLast(Integer.parseInt(arr[i]));

            // 명령 ㄱㄱ
            boolean first = true;
            for (int i = 0; i < p.length(); i++) {
                char command = p.charAt(i);
                if (command == 'R') {
                    first = !first;
                } else {
                    // 배열이 비어있으면 에러
                    if (deque.isEmpty()) {
                        sb.append("error\n");
                        continue out;
                    }
                    // first가 true라면 앞에부터 제거
                    else if (first) {
                        deque.removeFirst();
                    } else {
                        deque.removeLast();
                    }
                }
            }

            if (deque.isEmpty()) {
                sb.append("[]\n");
                continue;
            }

            sb.append("[");

            if (first) {
                while (!deque.isEmpty()) {
                    sb.append(deque.removeFirst()).append(",");
                }
            } else {
                while (!deque.isEmpty()) {
                    sb.append(deque.removeLast()).append(",");
                }
            }
            sb.deleteCharAt(sb.length() - 1);
            sb.append("]\n");
        }
        System.out.print(sb);
    }
}