Merge pull request TheAlgorithms#939 from 0Zeta/0zeta/prime-check-correction

yanglbme · web-flow · commit b8b293855c34 · 2019-10-02T16:40:42.000+08:00
Correct prime check
diff --git a/Maths/PrimeCheck.java b/Maths/PrimeCheck.java
@@ -21,6 +21,12 @@ public static void main(String[] args) {
      * @return {@code true} if {@code n} is prime
      */
     public static boolean isPrime(int n) {
+        if (n == 2) {
+            return true;
+        }
+        if (n < 2 || n % 2 == 0) {
+            return false;
+        }
         for (int i = 3; i <= Math.sqrt(n); i += 2) {
             if (n % i == 0) {
                 return false;
