[JAVA] 백준 (골드5) 5430번 AC

링크

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

---

문제 설명

정답률 20.314%

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

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

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

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

---

입력 예제

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

---

풀이

구현 자체는 간단한 문제이나 단순 구현만 하면 시간 초과가 발생하기 쉽다. 우선 시간 복잡도가 많이 발생할 수 있는 부분은 다음과 같다.

• 배열 뒤집기
• 첫번째 원소 삭제

List로 숫자를 저장할 경우 리스트를 뒤집을 때(Collections.reverse(list)) 시간 복잡도는 $O(n)$이고, 첫번째 원소를 삭제할 경우 나머지 원소들이 이동해야 하므로 시간 복잡도는 $O(n)$이다. R과 D 명령어의 개수가 m일 때 결국 시간 복잡도는 $O(n*m)$이 된다.

이때 Deque을 사용하여 시간 복잡도를 줄일 수 있다. 배열을 실제로 뒤집지 않고 뒤집기 여부를 를 변수로 저장하고, 정방향이면 removeFirst(), 역방향이면 removeLast()를 이용하여 원소를 삭제하면 명령어 당 시간 복잡도는 $O(1)$이 되고 최종적으로 $O(m)$ 수준으로 줄일 수 있다.

전체 코드

//백준
public class Main {

    public static void main(String[] args) throws IOException {

        System.setIn(new FileInputStream("src/input.txt"));
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int T = Integer.parseInt(br.readLine());

        for (int testCase = 0; testCase < T; testCase++) {
            String[] split = br.readLine().split("");
            int n = Integer.parseInt(br.readLine());
            String s = br.readLine();
            String result = "";

            Deque<String> deque = new ArrayDeque<>();
            if (n != 0) {
                String substring = s.substring(1, s.length() - 1);
                for (String num : substring.split(",")) {
                    deque.add(num);
                }
            }

            boolean isReversed = false;  //뒤집기 여부 저장
            for (String str : split) {
                if (str.equals("R")) {
                    isReversed = true;
                } else {
                    if (deque.isEmpty()) {
                        result = "error";
                        break;
                    }
                    if (isReversed) {
                        deque.removeLast();  //뒤집힌 상태면 뒤에서 제거
                    } else {
                        deque.removeFirst();  //정방향이면 앞에서 제거
                    }
                }
            }

            if (result.equals("error")) {
                System.out.println(result);
            } else {
                StringBuilder sb = new StringBuilder("[");
                while (!deque.isEmpty()) {
                    if (isReversed) {
                        sb.append(deque.removeLast());  //뒤집힌 상태면 뒤에서부터 출력
                    } else {
                        sb.append(deque.removeFirst());  //정방향이면 앞에서부터 출력
                    }
                    if (!deque.isEmpty()) {
                        sb.append(",");
                    }
                }
                sb.append("]");
                System.out.println(sb);
            }
            
        }
    }
}