/
ch-2.pl
executable file
·69 lines (47 loc) · 1.15 KB
/
ch-2.pl
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
#!/usr/bin/perl -s
use v5.24;
use Test2::V0 '!float';
use PDL;
use PDL::NiceSlice;
our ($tests, $examples);
run_tests() if $tests || $examples; # does not return
die <<EOS unless @ARGV;
usage: $0 [-examples] [-tests] [--] [M]
-examples
run the examples from the challenge
-tests
run some tests
M
a matrix in any format as accepted by the PDL string constructor,
e.g. '1,0,0;0,1,0;0,0,1'
EOS
### Input and Output
say submatrix_sum("@ARGV");
### Implementation
sub submatrix_sum {
my $m = pdl @_;
$m->range(ndcoords($m(1:,1:)), 2)->reorder(2, 3, 0, 1)
->clump(2)->sumover;
}
### Examples and tests
sub run_tests {
SKIP: {
skip "examples" unless $examples;
is submatrix_sum(
[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12])->unpdl,
[[14, 18, 22], [30, 34, 38]], 'example 1';
is submatrix_sum(
[1, 0, 0, 0],
[0, 1, 0, 0],
[0, 0, 1, 0],
[0, 0, 0, 1])->unpdl,
[[2, 1, 0], [1, 2, 1], [0, 1, 2]], 'example 2';
}
SKIP: {
skip "tests" unless $tests;
}
done_testing;
exit;
}