32 lines
967 B
Raku
32 lines
967 B
Raku
use Prime::Factor;
|
|
|
|
sub mo-prime($a, $p, $e) {
|
|
my $m = $p ** $e;
|
|
my $t = ($p - 1) * ($p ** ($e - 1)); # = Phi($p**$e) where $p prime
|
|
my @qs = 1;
|
|
for prime-factors($t).Bag -> $f {
|
|
@qs = flat @qs.map(-> $q { (0..$f.value).map(-> $j { $q * $f.key ** $j }) });
|
|
}
|
|
|
|
@qs.sort.first: -> $q { expmod( $a, $q, $m ) == 1 };
|
|
}
|
|
|
|
sub mo($a, $m) {
|
|
$a gcd $m == 1 or die "$a and $m are not relatively prime";
|
|
[lcm] flat 1, prime-factors($m).Bag.map: { mo-prime($a, .key, .value) };
|
|
}
|
|
|
|
multi MAIN('test') {
|
|
use Test;
|
|
|
|
for (10, 21, 25, 150, 1231, 123141, 34131) -> $n {
|
|
is ([*] prime-factors($n).Bag.map( { .key ** .value } )), $n, "$n factors correctly";
|
|
}
|
|
|
|
is mo(37, 1000), 100, 'mo(37,1000) == 100';
|
|
my $b = 10**20-1;
|
|
is mo(2, $b), 3748806900, 'mo(2,10**20-1) == 3748806900';
|
|
is mo(17, $b), 1499522760, 'mo(17,10**20-1) == 1499522760';
|
|
$b = 100001;
|
|
is mo(54, $b), 9090, 'mo(54,100001) == 9090';
|
|
}
|