[실버 4] 4134번 다음 소수

문제 링크 : https://www.acmicpc.net/problem/4134

문제

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

입력

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

출력

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

예제 입력 1
3
6
20
100

예제 출력 1
7
23
101

코드

• BigInteger 사용

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

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		int n = Integer.parseInt(br.readLine());
		
		for(int i=0; i<n; i++) {
			BigInteger num = new BigInteger(String.valueOf(Long.parseLong(br.readLine())));
			
			if(num.isProbablePrime(10)) {  // 현재 값이 소수인지 아닌지 판단하는 메소드
				System.out.println(num);
			} else {
				System.out.println(num.nextProbablePrime());  // 다음 소수 값을 반환해주는 메소드
			}
		}
	}
}

 

---

• long 사용

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

public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        
        StringBuilder sb = new StringBuilder();
        long[] datas = new long[N];
        for (int i=0; i<N; i++) {
            datas[i] = Long.parseLong(br.readLine());
        }
        
        for(int i=0; i<datas.length; i++) {
            long data = datas[i];
            while(!(isPrime(data))) {
                data++;
            }
            sb.append(data + "\n");
        }
        
        System.out.println(sb);
    }
    
    private static boolean isPrime(long number) {
        if (number < 2) {
            return false;
        }

        if (number == 2) {
            return true;
        }
        for (int i=2; i<=Math.sqrt(number); i++) {
            if (number % i == 0) {
                return false;
            }
        }
        
        return true;
    }
}