aboutsummaryrefslogtreecommitdiff
path: root/challenge-059/arne-sommer/raku/bit-sum-musing2
blob: 9562349691ba07aa2b913534b83ed072ffd84156 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#! /usr/bin/env raku

unit sub MAIN (Int $limit where $limit > 1 = 100, :$verbose, :$type = "int");

my $prev-sum = 0;
my $prev-inc = 0;
my $prev-inx = 0;

my @result;

for 2 .. $limit -> $number
{
  my @list;

  given $type
  {
    when "int"   { @list = (1     .. Inf)[^$number] }
    when "even"  { @list = (2, 4 ... Inf)[^$number] }
    when "odd"   { @list = (1, 3 ... Inf)[^$number] }
    when "prime" { @list = ((1 .. Inf).grep: *.is-prime)[^$number] }
    when "fib"   { @list = (1, 1, * + * ... Inf)[^$number] }
    default      { die "Unknown type $_" }
  }

  my $sum  = bit-sum(@list);
  my $inc  = $sum - $prev-sum;
  my $inx  = $inc - $prev-inc;
  
  say ": bit-sum @list[] -> { $sum.fmt("%3d") } ->  { $inc.fmt("%3d") } ->  { $inx.fmt("%3d") }" if $verbose;

  @result.push: $inx;
  
  $prev-sum = $sum;
  $prev-inc = $inc;
  $prev-inx = $inx;
}

put @result;

sub bit-sum (*@numbers where @numbers.elems > 1 && all(@numbers) ~~ Int)
{
  my $grand-total;

  for @numbers.combinations(2) -> $list
  {
    my $sum = bit-diff(|$list);
    $grand-total += $sum;
  }

  return $grand-total;

  sub bit-diff (Int $a, Int $b)
  {
    return ($a +^ $b).base(2).comb.sum;
  }
}