Source

project-euler / project-euler / 148 / calc-num-Y-in-row-ORIG-2.pl

Full commit
#!/usr/bin/perl

use strict;
use warnings;

use 5.016;

use integer;
# use Math::BigInt lib => 'GMP', ':constant';

my $BASE = 7;

sub calc_num_Y_in_row_n
{
    my $n_proto = shift;
    my $n = $n_proto - 1;

    my @D;

    # my $digit_n = $n->copy();
    my $digit_n = $n;
    my $power = 1;
    my $total_mod = 0;
    while ($digit_n)
    {
        my $digit = ($digit_n % $BASE);
        $total_mod = $total_mod + $digit * $power;
        push @D, { d => $digit, power => $power, total_mod => $total_mod };
        $digit_n /= $BASE;
        $power *= $BASE;
    }

    my $recurse = sub {
        my ($d_len) = @_;

        if ($d_len <= 0)
        {
            return 0;
        }
        else
        {
            my $big_Y_num = ($D[$d_len]->{power}-1-$D[$d_len-1]->{total_mod});
            my $big_Y_total = $big_Y_num * $D[$d_len]->{d};

            return $big_Y_total + ($D[$d_len]->{d}+1) * __SUB__->($d_len-1);
        }
    };

    return $recurse->($#D);
}

sub calc_num_Y_in_7_rows
{
    my $n_proto = shift;
    my $n = $n_proto - 1;

    my @D;

    # my $digit_n = $n->copy();
    my $digit_n = $n;
    my $power = 1;
    my $total_mod = 0;
    while ($digit_n)
    {
        my $digit = ($digit_n % $BASE);
        $total_mod = $total_mod + $digit * $power;
        push @D, { d => $digit, power => $power, total_mod => $total_mod };
        $digit_n /= $BASE;
        $power *= $BASE;
    }

    if ($D[0]{d} != 0)
    {
        die "Cannot proceeed with '$n'.";
    }

    my $recurse = sub {
        my ($d_len) = @_;

        if ($d_len <= 0)
        {
            return 0;
        }
        else
        {
            my $big_Y_num = ($D[$d_len]->{power}-1-$D[$d_len-1]->{total_mod});
            my $big_Y_total = $big_Y_num * $D[$d_len]->{d};

            return $big_Y_total + ($D[$d_len]->{d}+1) * __SUB__->($d_len-1);
        }
    };

    return $recurse->($#D);
}

if ($ENV{RUN})
{
    my $n = 1_000_000_000;

    my $sum = 0;

    for (; $n >= 6; $n--)
    {
        $sum += calc_num_Y_in_row_n($n);

        if ($n % 10_000 == 0)
        {
            print "Reached $n [Sum == $sum]\n";
        }
    }
    print "Final Sum == $sum\n";
}
else
{
    foreach my $n (@ARGV)
    {
        print "${n}: ", calc_num_Y_in_row_n($n), "\n";
    }
}