46 lines
1.4 KiB
Perl
46 lines
1.4 KiB
Perl
|
:- use_module(library(pio)).
|
||
|
:- use_module(library(clpfd)).
|
||
|
:- initialization(main, main).
|
||
|
|
||
|
main([FileName|_]) :-
|
||
|
input(FileName, Input),
|
||
|
transpose(Input, Rotated),
|
||
|
concurrent_maplist([Row, [#|Row]]>>(true), Rotated, Rocks),
|
||
|
concurrent_maplist(row_weight, Rocks, Weights),
|
||
|
sum_list(Weights, W),
|
||
|
writef('%t\n', [W]).
|
||
|
|
||
|
row_weight(Row, Weight) :-
|
||
|
abolish_private_tables, table(rocks/3),
|
||
|
phrase(rocks(Rocks), Row),
|
||
|
convlist([N-'#', N]>>(true), Rocks, RockCounts),
|
||
|
sum_list(RockCounts, Weight),
|
||
|
writef('>> %t <-- %t\n', [Weight, Row]).
|
||
|
|
||
|
rocks([0-0]) --> eos.
|
||
|
rocks([LastN-I, LastN-LastI|Rocks]) -->
|
||
|
[0], rocks([LastN-LastI|Rocks]),
|
||
|
{I is LastI + 1}.
|
||
|
rocks([N-I, LastN-LastI|Rocks]) -->
|
||
|
[1], rocks([LastN-LastI|Rocks]),
|
||
|
{I is LastI + 1, N is LastN + 1}.
|
||
|
rocks([0-I, 0-LastI|Rocks]) -->
|
||
|
['#'], rocks([0-LastI|Rocks]),
|
||
|
{I is LastI + 1}.
|
||
|
rocks([0-I, N-'#', LastN-LastI|Rocks]) -->
|
||
|
['#'], rocks([LastN-LastI|Rocks]),
|
||
|
{LastN =\= 0, I is LastI + 1, N is LastN*(2*LastI - LastN + 1) / 2}.
|
||
|
|
||
|
% things below are for reading from input file
|
||
|
input(FileName, Input) :- phrase_from_file(lines(Input), FileName).
|
||
|
|
||
|
lines([]) --> eos, !.
|
||
|
lines([Line|Lines]) --> line(Line), lines(Lines).
|
||
|
|
||
|
line([]) --> "\n"; eos.
|
||
|
line([1|Chars]) --> "O", line(Chars).
|
||
|
line([0|Chars]) --> ".", line(Chars).
|
||
|
line(['#'|Chars]) --> "#", line(Chars).
|
||
|
|
||
|
eos([], []).
|