blob: a2d04d97dcef505cb3b692de966be4e689e5ec26 (
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
95
96
97
98
|
-- Perl Weekly Challenge 169
-- Task 1
CREATE SCHEMA IF NOT EXISTS pwc169;
CREATE OR REPLACE FUNCTION
pwc169.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;
CREATE OR REPLACE FUNCTION
pwc169.compute_factors( n int )
RETURNS SETOF INT
AS $CODE$
DECLARE
i int;
BEGIN
FOR i IN 2 .. ( n - 1 ) LOOP
IF NOT pwc169.is_prime( i ) THEN
CONTINUE;
END IF;
WHILE n % i = 0 LOOP
RETURN NEXT i;
n = n / i;
END LOOP;
END LOOP;
RETURN;
END
$CODE$
LANGUAGE plpgsql;
CREATE OR REPLACE FUNCTION
pwc169.task1_plpgsql( n int DEFAULT 20 )
RETURNS SETOF INT
AS $CODE$
DECLARE
i int;
f int;
current_length int := 0;
current_count int := 0;
ok bool := false;
previous_f int := 0;
BEGIN
FOR i IN 2 .. 100000 LOOP
current_length := 0;
current_count := 0;
ok := true;
previous_f := 0;
FOR f IN SELECT * FROM pwc169.compute_factors( i ) ORDER BY 1 LOOP
current_count := current_count + 1;
IF current_length = 0 THEN
current_length := length( f::text );
END IF;
IF length( f::text ) <> current_length OR current_count > 2 OR f = previous_f THEN
ok := false;
EXIT;
END IF;
previous_f := f;
END LOOP;
IF ok AND previous_f <> 0 THEN
RETURN NEXT i;
IF n = 0 THEN
RETURN;
END IF;
n := n - 1;
END IF;
END LOOP;
RETURN;
END
$CODE$
LANGUAGE plpgsql;
|
