Skip to content

Commit a170343

Browse files
committed
lc: add solution for finding closest prime numbers in a range
1 parent 6921253 commit a170343

1 file changed

Lines changed: 70 additions & 0 deletions

File tree

Lines changed: 70 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,70 @@
1+
package com.thealgorithms.leetcode;
2+
3+
import java.util.Arrays;
4+
5+
public class ClosestPrimeNumbersInRange {
6+
public int[] closestPrimes(int left, int right) {
7+
// get all primes up to right
8+
boolean[] isPrime = sieveOfEratosthenes(right);
9+
10+
// first two primes in range
11+
int first = -1;
12+
int second = -1;
13+
14+
int minDiff = Integer.MAX_VALUE;
15+
int prev = -1;
16+
17+
for (int i = Math.max(2, left); i <= right; i++) {
18+
if (isPrime[i]) {
19+
if (prev == -1) {
20+
prev = i;
21+
} else {
22+
int diff = i - prev;
23+
if (diff < minDiff) {
24+
minDiff = diff;
25+
first = prev;
26+
second = i;
27+
}
28+
prev = i;
29+
}
30+
}
31+
}
32+
33+
return first == -1 ? new int[]{-1, -1} : new int[]{first, second};
34+
}
35+
36+
public boolean isPrime(int n) {
37+
if (n <= 1) return false;
38+
if (n <= 3) return true;
39+
if (n % 2 == 0 || n % 3 == 0) return false;
40+
41+
// check for prime by testing divisors up to square root of n, can skip even numbers
42+
// and optimize by checking only numbers of form 6k + 1
43+
for (int i = 5; i * i <= n; i+= 6) {
44+
if (n % i == 0 || n % (i + 2) == 0) {
45+
return false;
46+
}
47+
}
48+
49+
return true;
50+
}
51+
52+
// helper method to find prime numbers using sieve of eratosthenes
53+
public static boolean[] sieveOfEratosthenes(int n) {
54+
boolean[] isPrime = new boolean[n+1];
55+
Arrays.fill(isPrime, true);
56+
isPrime[0] = false;
57+
isPrime[1] = false;
58+
59+
for (int i = 2; i * i <= n; i++) {
60+
if (isPrime[i]) {
61+
for (int j = i * i; j <= n; j += i) {
62+
isPrime[j] = false;
63+
}
64+
}
65+
}
66+
return isPrime;
67+
}
68+
}
69+
70+
// https://leetcode.com/problems/closest-prime-numbers-in-range/description/

0 commit comments

Comments
 (0)