blob: 0260bb29b339abb16bdcb30b9d4673687084c949 (
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
|
package PWC262;
/**
* PL/Java implementation for PWC 262
* Task 2
* See <https://perlweeklychallenge.org/blog/perl-weekly-challenge-262>
*
*
* 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/PWC262-1.jar luca@rachel:/tmp
* To install into PostgreSQL execute:
select sqlj.install_jar( 'file:///tmp/PWC262-1.jar', 'PWC262', true );
select sqlj.set_classpath( 'public', 'PWC262' );
select pwc262.task2_pljava();
and then to redeploy:
select sqlj.replace_jar( 'file:///tmp/PWC262-1.jar', 'PWC262', 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 Task2 {
private final static Logger logger = Logger.getAnonymousLogger();
@Function( schema = "pwc262",
onNullInput = RETURNS_NULL,
effects = IMMUTABLE )
public static final int task2_pljava( int k, int[] nums ) throws SQLException {
logger.log( Level.INFO, "Entering pwc262.task2_pljava" );
int pairs = 0;
for ( int i = 0; i < nums.length; i++ )
for ( int j = i + 1; j < nums.length; j++ )
if ( nums[ i ] != nums[ j ]
|| ( i * j ) % k != 0 )
continue;
else
pairs++;
return pairs;
}
}
|