blob: 9e27f788b2e562e1662fd2a70e72c68f3eefb2e2 (
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
|
-- Perl Weekly Challenge 164
-- Task 2
CREATE SCHEMA IF NOT EXISTS pwc164;
CREATE OR REPLACE FUNCTION
pwc164.is_happy( n int )
RETURNS bool
AS $CODE$
DECLARE
s int;
nn int;
BEGIN
WHILE n > 10 LOOP
s := 0;
FOREACH nn IN ARRAY regexp_split_to_array( n::text, '' )::int[] LOOP
s := s + nn * nn;
END LOOP;
n := s;
END LOOP;
IF n = 1 THEN
RETURN true;
ELSE
RETURN false;
END IF;
END
$CODE$
LANGUAGE plpgsql;
CREATE OR REPLACE FUNCTION
pwc164.task2_plpgsql( l int DEFAULT 8 )
RETURNS SETOF INT
AS $CODE$
DECLARE
BEGIN
FOR n IN 10 .. 99999 LOOP
IF pwc164.is_happy( n ) THEN
RETURN NEXT n;
l := l - 1;
END IF;
EXIT WHEN l = 0;
END LOOP;
RETURN;
END
$CODE$
LANGUAGE plpgsql;
|
