Skip to content

Instantly share code, notes, and snippets.

@felher
Created January 16, 2012 19:06
Show Gist options
  • Save felher/1622404 to your computer and use it in GitHub Desktop.
Save felher/1622404 to your computer and use it in GitHub Desktop.
use v6;
# the modexp-function from: http://rosettacode.org/wiki/Modular_exponentiation
sub modexp(Int $a is copy, Int $b is copy, $n) {
my $c = 1;
repeat while $b div= 2 {
($c *= $a) %= $n if $b % 2;
($a *= $a) %= $n;
}
$c;
}
subset PrimeCandidate of Int where { $_ > 2 and $_ % 2 };
my Bool multi sub is_prime(Int $n, Int $k) { return False; }
my Bool multi sub is_prime(2, Int $k) { return True; }
my Bool multi sub is_prime(PrimeCandidate $n, Int $k) {
my Int $d = $n - 1;
my Int $s = 0;
while $d %% 2 {
$d div= 2;
$s++;
}
for (2 ..^ $n).pick($k) -> $a {
my $x = modexp($a, $d, $n);
next if $x == 1 or $x == $n - 1;
for 1 ..^ $s {
$x = $x ** 2 mod $n;
return False if $x == 1;
last if $x == $n - 1;
}
return False if $x !== $n - 1;
}
return True;
}
say (1..1000).grep({ is_prime($_, 10) }).join(", ");
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment