aboutsummaryrefslogtreecommitdiff
path: root/challenge-158/luca-ferrari/postgresql/ch-1.sql
blob: 4aaff9fc622bd9197f3f3ce852c3aae75013befb (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
CREATE SCHEMA IF NOT EXISTS pwc158;

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

RETURN true;
END
$CODE$
LANGUAGE plpgsql;


CREATE OR REPLACE FUNCTION
pwc158.sum_digits( n bigint )
RETURNS int
AS $CODE$
   WITH s(i) AS ( SELECT regexp_split_to_table( n::text, '' ) )
   SELECT sum( i::bigint ) FROM s;
$CODE$
LANGUAGE SQL;


SELECT n
FROM generate_series( 1, 100 ) n
WHERE pwc158.is_prime( n )
AND pwc158.is_prime( pwc158.sum_digits( n ) );