코린이의 소소한 공부노트

[백준 온라인 저지] 4134. 다음 소수 본문

코딩테스트 풀이/JAVA

[백준 온라인 저지] 4134. 다음 소수

무지맘 2023. 5. 2. 20:57

1. 입력

- 첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 정수 n(0 n 4*10^9)이 주어진다.

 

2. 출력

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

 

3. 코드

import java.util.*;
import java.io.*;
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 n = Integer.valueOf(br.readLine());
        for(int i=0 ; i<n ; i++){
            long num = Long.valueOf(br.readLine());
            bw.write(findPN(num)+"\n");
        }
        bw.flush();
        bw.close();
    }
    static long findPN(long a){
        if(a<=2) return 2;
        boolean find = false;
        while(!find){
            boolean isPrime = true;
            for(long i=2 ; i<=Math.sqrt(a) ; i++)
                if(a%i==0) {
                    isPrime = false; break;
                }
            if(isPrime)
                find = true;
            else
                a++;
        }
        return a;
    }
}