[BOJ][JAVA] 5430 AC

5430 AC

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

문제
선영이는 주말에 할 일이 없어서 새로운 언어 AC를 만들었다. AC는 정수 배열에 연산을 하기 위해 만든 언어이다. 이 언어에는 두 가지 함수 R(뒤집기)과 D(버리기)가 있다. 함수 R은 배열에 있는 수의 순서를 뒤집는 함수이고, D는 첫 번째 수를 버리는 함수이다. 배열이 비어있는데 D를 사용한 경우에는 에러가 발생한다. 함수는 조합해서 한 번에 사용할 수 있다. 예를 들어, "AB"는 A를 수행한 다음에 바로 이어서 B를 수행하는 함수이다. 예를 들어, "RDD"는 배열을 뒤집은 다음 처음 두 수를 버리는 함수이다. 배열의 초기값과 수행할 함수가 주어졌을 때, 최종 결과를 구하는 프로그램을 작성하시오.

입력
첫째 줄에 테스트 케이스의 개수 T가 주어진다. T는 최대 100이다. 각 테스트 케이스의 첫째 줄에는 수행할 함수 p가 주어진다. p의 길이는 1보다 크거나 같고, 100,000보다 작거나 같다. 다음 줄에는 배열에 들어있는 수의 개수 n이 주어진다. (0 ≤ n ≤ 100,000) 다음 줄에는 [x1,...,xn]과 같은 형태로 배열에 들어있는 정수가 주어진다. (1 ≤ xi ≤ 100) 전체 테스트 케이스에 주어지는 p의 길이의 합과 n의 합은 70만을 넘지 않는다.

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

풀이 방법

오랜만에 푸는 골드 문제라 겁먹었는데 생각보다 간단한 논리로 해결됐다.

우선 해당 문제는 최대 10만 길이의 배열을 최대 10만 번 처리를 해야 하고 최대 100번 까지 반복해야 한다.
즉 T가 100, n이 10만, p가 R이 10만 개일 경우 10만 길이의 배열을 10만 번 reverse하는 과정을 100번 반복해야 한다는 것이다.

최악의 조건이지만 문제의 시간 제한이 1초인 점을 고려하면 최악의 조건에서도 1초 안에 돌아가야 한다는 뜻이므로 단순한 배열 뒤집기, 삭제하기로는 해결할 수 없는 문제이다.

나는 그래서 배열을 직접 뒤집는 것이 아니라 index를 뒤집기로 했다.
index 값만 조정하면 실제 배열의 값은 바뀌지 않으면서 매우 빠르게 뒤집기, 삭제 연산을 처리할 수 있기 때문이다.

처음 배열이 입력되면 start, end 인덱스를 지정하고 방향을 설정한다.
방향은 현재 배열이 뒤집힌 상태인지(← 방향) 기본 상태인지(→ 방향)를 나타낸다.

D 입력에 대해서는 index와 방향에 따른 조건문을 만들어 줬는데

1. 방향이 정방향(→)이며 start가 end 이하인 경우
2. 방향이 역방향(←)이며 start가 end 이상인 경우

두 경우만 연산 처리가 가능한 상황이기 때문에 이외의 경우에서는 error 처리를 해주었다.
정방향인데 start가 end보다 크거나 역방향인데 start가 end보다 작은 경우는 배열이 비어있는데 삭제 연산을 하려는 경우이기 때문이다.

연산에 따른 index 계산이 끝나고 나면 또다시 방향에 따라 수를 배열에서 꺼내 출력해주었다.

예를 들어 연산이 RDD, 숫자가 [1,2,3,4,5]가 입력됐다고 해보자.

이 상태에서 R을 하면 방향은 역방향으로 바뀌고 start와 end가 뒤바뀔 것이다.

이후 D가 입력됐으므로 현재 상태인 [5,4,3,2,1]에서 맨 앞 5가 삭제돼야 한다.
즉 start가 한 칸 앞으로 땡겨져야 하는 것이다.

그 다음에도 D가 입력됐으므로 start를 한 칸 앞으로 땡겨준다.

최종적으로 [3,2,1]이 출력될 것이다.
실제로 계산을 해봐도 [1,2,3,4,5] -R-> [5,4,3,2,1] -D-> [4,3,2,1] -D-> [3,2,1] 이 된다.

이렇게 index만 조정해주고 마지막에 배열에서 숫자를 뽑아내기만 하는 방법으로 해결할 수 있었다.

소스 코드

import java.util.Scanner;

public class Problem_5430 {
    public static void main(String[] args) throws Exception {
        Scanner scanner = new Scanner(System.in);
        StringBuilder sb = new StringBuilder();

        int n = scanner.nextInt();
        scanner.nextLine();

        for (int i = 0; i < n; i++) {
            String operation = scanner.nextLine();

            int start = 0;
            int max = scanner.nextInt();
            int end = max - 1;
            scanner.nextLine();

            // 방향(정방향인지 역방향인지)
            Character dir = 'R';

            Boolean flag = true;

            String input = scanner.nextLine();
            input = input.substring(1, input.length() - 1);
            String[] numbers = input.split(",");

            for (int j = 0; j < operation.length(); j++) {
                if (operation.charAt(j) == 'R') {
                    int tmp = start;
                    start = end;
                    end = tmp;

                    switch (dir) {
                        case 'R':
                            dir = 'L';
                            break;
                        case 'L':
                            dir = 'R';
                            break;
                    }
                } else if (operation.charAt(j) == 'D') {
                    // 뒤집히지 않은 경우(정방향)
                    if (start <= end && dir == 'R') {
                        start++;
                    }
                    // 뒤집은 경우(역방향)
                    else if (start >= end && dir == 'L') {
                        start--;
                    } else {
                        flag = false;
                        break;
                    }
                }
            }

            if (!flag) {
                sb.append("error\n");
            } else {
                sb.append("[");

                int j;
                if (start <= end && dir == 'R') {
                    for (j = start; j < end; j++) {
                        sb.append(numbers[j] + ",");
                    }

                    sb.append(numbers[j]);
                } else if (start >= end && dir == 'L') {
                    for (j = start; j > end; j--) {
                        sb.append(numbers[j] + ",");
                    }

                    sb.append(numbers[j]);
                }

                sb.append("]\n");
            }
        }

        scanner.close();

        System.out.println(sb);
    }
}