[BaekJoon] 5430 AC (Java)

🔗 문제 링크

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

📝 문제풀이 방법

해당 문제는 조건에 따라 수행할 함수, 배열에 있는 수, 수의 배열을 입력받으며 문제를 해결하여야 합니다.

1. 문자열에서 리스트로 파싱

수의 배열의 경우는 숫자가 아닌 [1,2,3,4]와 같은 형태로 주어져서 파싱하는 작업을 진행하며 내부의 숫자를 사용자의 문제풀이 방법에 따라 배열 또는 리스트에 담아주는 작업을 하여야합니다. 저는 뒤집는 명령을 Collections의 reverse()메서드를 통해 문제를 풀이하여 배열을 리스트로 반환하였습니다.

파싱 부분의 작성 코드는 parseArr()메서드를 통해 확인하실 수 있습니다.

2. 명령 함수 수행

파싱을 한 후에는 R과 D로 이루어진 수행할 함수의 명령에 따른 수행 메서드를 구현해야합니다. 명령에 따라 각각 list를 뒤집거나 값을 삭제하는 작업을 하였습니다. 여기서 값의 삭제시 값의 부족으로 인한 오류는 try/catch문으로 IndexOutOfBoundsException을 잡을 시 error를 리턴하도록 작성하였습니다.

명령 수행을 할 때 R 명령이 나올 때마다 Collections.reverse()메서드를 통해 뒤집게 된다면 내부적으로 많은 시간이 소모되어서 코드 제출 시 실행초과라는 결과를 받게 됩니다. 그래서 이 부분은 isReverse변수를 따로 생성하여 Collections.reverse()는 딱 한번만 수행되도록 하였더니 코드가 통과를 하게 되었습니다

명령 수행 코드는 executeFunction()메서드를 통해 확인하실 수 있습니다.

3. 리스트 결과를 문자열로 변환

명령 수행을 마친 후에는 남은 배열을 옳바른 형태로 다시 변환하여 String타입으로 return하였습니다.

결과를 String으로 파싱하는 코드는 makeResultString()메서드를 통해 확인하실 수 있습니다.

👨🏻‍💻 작성한 코드

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

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

        for (int i = 0; i < T; i++) {
            String command = br.readLine();
            int n = Integer.parseInt(br.readLine());
            String inputArr = br.readLine();

            bw.append(executeFunction(command, n, parseArr(inputArr)));
        }

        bw.flush();
        bw.close();
        br.close();
    }

    public static List<Integer> parseArr(String inputArr) {
        List<Integer> numList = new ArrayList<>();
        Arrays.stream(inputArr.substring(1, inputArr.length() - 1).split(","))
                .forEach(s -> {
                    if (!s.equals("")) numList.add(Integer.parseInt(s));
                });
        return numList;
    }

    public static String executeFunction(String command, int n, List<Integer> numList) {
        boolean isReverse = false;
        for (int i = 0; i < command.length(); i++) {
            if (command.charAt(i) == 'R') {
                isReverse = !isReverse;
            }

            if (command.charAt(i) == 'D') {
                try {
                    if (isReverse)
                        numList.remove(numList.size() - 1);
                    else
                        numList.remove(0);
                } catch (IndexOutOfBoundsException e) {
                    return "error\n";
                }
            }
        }

        if (isReverse) {
            Collections.reverse(numList);
        }

        return makeResultString(numList);
    }

    public static String makeResultString(List<Integer> numList) {
        if (numList.isEmpty()) {
            return "[]\n";
        }

        StringBuilder sb = new StringBuilder();
        sb.append("[");
        numList.stream()
                .forEach(integer -> {
                    sb.append(integer);
                    sb.append(",");
                });
        sb.deleteCharAt(sb.length() - 1);
        sb.append("]\n");
        return sb.toString();
    }
}