blob: 5a49f2024ddebaea1e0f8b2a1aacb43e8cb0950c (
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
99
|
package PWC270;
/**
* PL/Java implementation for PWC 270
* Task 1
* See <https://perlweeklychallenge.org/blog/perl-weekly-challenge-270>
*
*
* To compile on the local machine:
$ export JAVA_HOME=/usr/lib/jvm/java-17-openjdk-amd64/ # if not already set
$ mvn clean build
$ scp target/PWC270-1.jar luca@rachel:/tmp
* To install into PostgreSQL execute:
select sqlj.install_jar( 'file:///tmp/PWC270-1.jar', 'PWC270', true );
select sqlj.set_classpath( 'public', 'PWC270' );
select pwc270.task2_pljava();
and then to redeploy:
select sqlj.replace_jar( 'file:///tmp/PWC270-1.jar', 'PWC270', true );
*/
import org.postgresql.pljava.*;
import org.postgresql.pljava.annotation.Function;
import static org.postgresql.pljava.annotation.Function.Effects.IMMUTABLE;
import static org.postgresql.pljava.annotation.Function.OnNullInput.RETURNS_NULL;
import java.util.*;
import java.util.stream.*;
import java.sql.SQLException;
import java.util.logging.*;
import java.sql.ResultSet;
import java.sql.Date;
public class Task1 {
private final static Logger logger = Logger.getAnonymousLogger();
@Function( schema = "pwc270",
onNullInput = RETURNS_NULL,
effects = IMMUTABLE )
public static final String[] task1_pljava( int row_size, int[] matrix ) throws SQLException {
logger.log( Level.INFO, "Entering pwc270.task1_pljava" );
List<Integer> rows = new LinkedList<Integer>();
List<Integer> cols = new LinkedList<Integer>();
List<String> result = new LinkedList<String>();
// seek for ones
for( int row = 0; row < ( matrix.length / row_size ) ; row++ )
for ( int col = 0; col < row_size; col++ )
if ( matrix[ row * row_size + col ] == 1 ) {
rows.add( row );
cols.add( col );
}
for ( int i = 0; i < rows.size(); i++ ) {
int current_row = rows.get( i );
int current_col = rows.get( i );
boolean ok = true;
for ( int r = 0; r < rows.size(); r++ )
if ( r == i )
continue;
else if ( rows.get( r ) == current_row )
ok = false;
for ( int c = 0; c < cols.size(); c++ )
if ( c == i )
continue;
else if ( cols.get( c ) == current_col )
ok = false;
if ( ok )
result.add( String.format( "Row %d Col %d", current_row, current_col ) );
}
String[] to_return = new String[ result.size() ];
int i = 0;
for ( String s : result )
to_return[ i++ ] = s;
return to_return;
}
}
|
