안녕하세요 강사님

예시 문제는 합격하는데 다른 4개의 테스트 케이스는 통과하지 못하고 있습니다.

문제는 런타임에러가 뜨는데 왜 뜨는지 정확한 이유를 아무리 보아도 모르겠습니다...

많이 더러운 코드지만 무엇이 문제인지 확인 한번 부탁 드리겠습니다...

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class Main {

    public String solution(String str) {
        String answer = "";
        String[] arr = new StringBuilder(str).reverse().toString().split(" ");
        List<Integer> list = new ArrayList<>();

        for (String s : arr) {
            int target = Integer.parseInt(s);

            if (target != 2 && target % 2 == 0) continue;

            if (target == 1) continue;
            boolean check = true;

            for (int j = 3; j <= Math.sqrt(target); j+=2) {
                if (target % j == 0) {
                    check = false;
                    break;
                }
            }
            if (check) list.add(target);
        }

        for (int i =list.size()-1; i >= 0; i--) {
            answer += list.get(i) + " ";
        }
        return answer;
    }

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int input1 = in.nextInt();
        in.nextLine();
        String str = in.nextLine();
        Main main = new Main();
        System.out.println(main.solution(str));
        return;
    }
}

알아낸점

저와 비슷한 사례를 겪는 분들을 질문창에서 발견하고 강사님이 적어두신 2번 케이스의 문제를 복사해서 적어보았는데 통과하였습니다.

저는 split() 메서드를 이용해서 입력받은 문자열을 배열로 변경하는데 이때 문자열 맨 뒤에 공백이 존재하게 되면 런타임 에러가 발생 하는것을 확인했습니다. 아무래도 그부분 때문인것 같은데 맞다면 확인 한번 부탁드리겠습니다.

변경

코드를 수정하였습니다.

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class Main {

    public String solution(String[] arr) {
        String answer = "";

        for (int i = 0; i < arr.length; i++) {
            String s = new StringBuilder(arr[i]).reverse().toString();

            int target = Integer.parseInt(s);

            if (target != 2 && target % 2 == 0) continue;

            if (target == 1) continue;

            boolean check = isPrime(target);

            if (check) answer += target + " ";
        }

        return answer;
    }

    private boolean isPrime(int target) {
        for (int j = 3; j <= Math.sqrt(target); j+=2) {
            if (target % j == 0) {
                return false;
            }
        }
        return true;
    }
    
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int input1 = in.nextInt();
        String[] arr = new String[input1];
        for (int i = 0; i < input1; i++) {
            arr[i] = in.next();
        }
        Main main = new Main();
        System.out.println(main.solution(arr));
        return;
    }
}

이렇게 하니 해결되었습니다.

궁금증

혹시 실제 코딩테스트 볼때도 split을 사용할때 요청값끝부분에 있는 공백의 존재 가능성도 염두해두고 사용을 해야하나요? 아니면 단순 이 사이트의 실수였을까요?