[JAVA] 백준 4134번 다음 소수

문제

 

정수 n(0 ≤ n ≤ 4*109)가 주어졌을 때, n보다 크거나 같은 소수 중 가장 작은 소수 찾는 프로그램을 작성하시오.

 

입력

 

첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 정수 n이 주어진다.

 

출력

 

각각의 테스트 케이스에 대해서 n보다 크거나 같은 소수 중 가장 작은 소수를 한 줄에 하나씩 출력한다.

 

입력예제

 

3
6
20
100

 

출력예제

 

7
23
101

 

풀이

import java.io.*;
import java.math.BigInteger;
import java.util.*;

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

        Long n = scanner.nextLong();
        Long[] arr = new Long[Math.toIntExact(n)];

        // 입력받기
        for (int i = 0; i < arr.length; i++) {
            arr[i] = scanner.nextLong();
            BigInteger bigint = BigInteger.valueOf(arr[i]);
            if (bigint.isProbablePrime(10)){
                bw.write(String.valueOf(bigint) + "\n");

            }else {
                bw.write(String.valueOf(bigint.nextProbablePrime()) + "\n");
            }
        }


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

 

 

.nextProbablePrime()은 다음 소수를 찾아주는 함수

.isProbablePrime() 소수인지 판별해주는 함수이다. → () 안에는 몇진수를 기준으로 할 지를 적는다.

 

 

 

 

 

4134번: 다음 소수