Sha256: f31da0461fddfc50e323973c2f6fb6a253990f0ff505a1d5a4f410e267b121d7
Contents?: true
Size: 1.38 KB
Versions: 116
Compression:
Stored size: 1.38 KB
Contents
type position = Start | UpperSide | RightSide | LowerSide | LeftSide
let count_rectangles pic =
let maxY = Array.length pic in
let maxX = if maxY > 0 then String.length pic.(0) else 0 in
let rec go x y xStart yStart pos =
let on_side_fn dx dy next_pos () =
go (x+dx) (y+dy) xStart yStart next_pos in
let on_corner_fn dx1 dy1 dx2 dy2 next_pos () =
go (x+dx1) (y+dy1) xStart yStart next_pos + go (x+dx2) (y+dy2) xStart yStart pos in
let progress side_ch on_side on_corner =
try
match pic.(y).[x] with
| c when c = side_ch -> on_side ()
| '+' -> on_corner ()
| _ -> 0
with Invalid_argument _ -> 0 in
match pos with
| Start ->
if y = maxY then 0
else if x = maxX then go 0 (y+1) 0 (y+1) Start
else (match pic.(y).[x] with
| '+' -> go (x+1) y x y UpperSide + go (x+1) y x y Start
| _ -> go (x+1) y (x+1) y Start
)
| UpperSide -> progress '-' (on_side_fn 1 0 UpperSide) (on_corner_fn 0 1 1 0 RightSide)
| RightSide -> progress '|' (on_side_fn 0 1 RightSide) (on_corner_fn (-1) 0 0 1 LowerSide)
| LowerSide -> progress '-' (on_side_fn (-1) 0 LowerSide) (on_corner_fn 0 (-1) (-1) 0 LeftSide)
| LeftSide -> progress '|' (on_side_fn 0 (-1) LeftSide) (fun () -> if x = xStart && y = yStart then 1 else go x (y-1) xStart yStart LeftSide)
in go 0 0 0 0 Start
Version data entries
116 entries across 116 versions & 1 rubygems