aboutsummaryrefslogtreecommitdiff
path: root/challenge-252/luca-ferrari/postgresql/ch-2.plperl
blob: a7abf562171e12c06df2d3b5d6497db075442df3 (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
--
-- Perl Weekly Challenge 252
-- Task 2
-- See <https://perlweeklychallenge.org/blog/perl-weekly-challenge-252/>
--

CREATE SCHEMA IF NOT EXISTS pwc252;

CREATE OR REPLACE FUNCTION
pwc252.task2_plperl( int )
RETURNS SETOF int
AS $CODE$
   my ( $size ) = @_;

   die "Cannot have a size less than 3!" if ( $size <= 3 );

   if ( $size % 2 == 0 ) {
      for ( 1 .. $size / 2 ) {
      	  return_next( $_ );
	  return_next( $_ * -1 );
      }
   }
   else {
       for ( 1 .. ( $size - 1 ) / 2 ) {
      	  return_next( $_ );
	  return_next( $_ * -1 );
      }

      my $next = int( $size / 2 ) + 1;
      return_next( $next );
      return_next( $next + 1 );
      return_next( ( $next + $next + 1 ) * -1 );
   }

return undef;
$CODE$
LANGUAGE plperl;