aboutsummaryrefslogtreecommitdiff
path: root/challenge-155/luca-ferrari/postgresql/ch-1.sql
blob: 421afabdbdc66aee98a15b3ee7a83adfc3bcdb09 (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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
CREATE SCHEMA IF NOT EXISTS pwc155;

/*
testdb=> select pwc155.fortunate( 8 ) order by 1;
fortunate
-----------
3
5
7
13
17
19
23
37
(8 rows)


*/

CREATE OR REPLACE FUNCTION
pwc155.fortunate( int )
RETURNS SETOF integer
AS $CODE$

# a subroutine to see if a number
# is prime
my $is_prime = sub {
   return 1 if $_[0] == 1;
   return 1 if $_[0] == 2;
   for my $i ( 2 .. $_[0] - 1 ) {
       return 0 if $_[0] % $i == 0;
   }

   return 1;
};

# generates the first n primes
my $generate_primes = sub {
   my @primes;
   for my $p ( 2 .. 99999 ) {
       push @primes, $p if $is_prime->( $p );
       return @primes if @primes == $_[0];
   }
};

my $max = sub {
   my $max = 0;
   for (@_) {
       $max = $_ if $_ > $max;
   }

   elog( DEBUG, "MAX = $max  in " . join( ',', @_ ) );
   return $max;
};


my $pn = sub {
   my $result = 1;
   for ( @_ ) {
       $result *= $_;
   }

   return $result;
};

my $limit = $_[0] || 8;
my %unique;

for my $n ( 1 .. 999999 ) {
    # generate the first n primes
    my @primes = $generate_primes->( $n );
    my $start  = $max->( @primes ) + 1;
    elog( DEBUG, "Primes = " . join( ',', @primes ) . " with max = $start" );
    for my $m ( $start .. 999999 ) {
         my $fortunate = $pn->( @primes ) + $m;
         elog( DEBUG, "Computing $m -> " . $pn->( @primes ) . " + $m = $fortunate = " . $is_prime->( $fortunate ) );
         next if ! $is_prime->( $fortunate );
         $unique{ $m }++;
         next if $unique{ $m } > 1;
         return_next( $m );
         last;
    }


    $limit--;
    last if $limit <= 0;

}



return undef;
$CODE$
LANGUAGE plperl;