aboutsummaryrefslogtreecommitdiff
path: root/challenge-154/luca-ferrari/postgresql/ch-2.sql
blob: 118285c1cee46da4331e66a45fcfc4fdeb9c1fcf (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
CREATE SCHEMA IF NOT EXISTS pwc154;

CREATE OR REPLACE FUNCTION
pwc154.padovan( i int )
RETURNS int
AS $CODE$
BEGIN
        IF i <= 2 THEN
           RETURN 1;
        END IF;

        RETURN pwc154.padovan( i - 3 ) + pwc154.padovan( i - 2 );
END
$CODE$
LANGUAGE plpgsql;

CREATE OR REPLACE FUNCTION
pwc154.is_prime( n int )
RETURNS bool
AS $CODE$
DECLARE
    i int;
BEGIN
    FOR i IN 2 .. n - 1 LOOP
        IF n % i = 0 THEN
           RETURN false;
        END IF;
    END LOOP;

    RETURN true;
END
$CODE$
LANGUAGE plpgsql;

/**
p
----
1
2
3
4
5
7
9
12
16
21
(10 rows)
*/
WITH
padovan AS (
   SELECT n, pwc154.padovan( n ) AS p
   FROM generate_series( 0, 50 ) n
)
, padovan_prime
AS (
  SELECT p, pwc154.is_prime( p ) AS prime
  FROM padovan p
)
SELECT distinct( p )
FROM padovan_prime
WHERE prime
ORDER BY 1
LIMIT 10
;