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

CREATE SCHEMA IF NOT EXISTS pwc215;

CREATE OR REPLACE FUNCTION
pwc215.task2_plperl( int, int[] )
RETURNS int
AS $CODE$
   my ( $count, $digits ) = @_;
   my ( $done );

   while ( $count ) {
   	 $done = 0;
	 for ( 1 .. scalar( $digits->@* ) - 1 ) {
	     if ( $digits->[ $_ ] == 0
	     	&& $digits->[ $_ - 1 ] == 0
		&& $digits->[ $_ + 1 ] == 0 ) {
		$digits->[ $_ ] = 1;
		$done = 1;
		$count--;
		last;
	     }
	 }

	 last if ! $done;
   }

   return 1 if ! $count;
   return 0;
$CODE$
LANGUAGE plperl;