fixed minor bug in prime testing java function

This commit is contained in:
William Ball 2020-07-17 22:41:46 -07:00
parent 0e62b02c92
commit 4abb208669
3 changed files with 10 additions and 8 deletions

View file

@ -6,14 +6,8 @@ public class Main
{ {
Util.init(); Util.init();
ArrayList<Long> vals = new ArrayList<Long>(); Tree tree = new Tree(0, new ArrayList<Long>());
vals.add(2l); for (int i = 0; i < 9; i++) {
vals.add(3l);
vals.add(5l);
vals.add(7l);
Tree tree = new Tree(0, vals);
for (int i = 0; i < 20; i++) {
tree.step(); tree.step();
Util.printList(tree.longestPath()); Util.printList(tree.longestPath());
} }

View file

@ -17,6 +17,10 @@ class Util
if (primes.containsKey(val)) { if (primes.containsKey(val)) {
return primes.get(val); return primes.get(val);
} }
if (val < 2) {
primes.put(val, false);
return false;
}
for (int i = 2; i * i <= val; i++) { for (int i = 2; i * i <= val; i++) {
if (val % i == 0) { if (val % i == 0) {
primes.put(val, false); primes.put(val, false);

View file

@ -21,6 +21,10 @@ class Util
if (primes.containsKey(val)) { if (primes.containsKey(val)) {
return primes.get(val); return primes.get(val);
} }
if (val < 2) {
primes.put(val, false);
return false;
}
for (int i = 2; i * i <= val; i++) { for (int i = 2; i * i <= val; i++) {
if (val % i == 0) { if (val % i == 0) {
primes.put(val, false); primes.put(val, false);