03a.exs

# elixir 03a.exs

defmodule Solution do
    def is_digit(x) do
        x in ?0..?9
    end

    def is_symbol(x) do
        !is_digit(x) && x != ?. && x != nil
    end

    def solve(a, x, y0, result) do
        if x == map_size(a) do
            result
        else
            m = map_size(a[x]) - 1
            y1 = Enum.find(y0..m, fn y -> is_digit(a[x][y]) end)
            if y1 == nil do
                solve(a, x + 1, 0, result)
            else
                y2 = Enum.find(y1..m, fn y -> !is_digit(a[x][y]) end) - 1
                number = List.foldl(Enum.to_list(y1..y2), 0, fn y, acc -> acc * 10 + (a[x][y] - 48) end)
                current = if Enum.any?((y1 - 1)..(y2 + 1), fn y ->
                    Enum.any?(max(x - 1, 0)..min(x + 1, map_size(a) - 1), fn x ->
                        is_symbol(a[x][y])
                    end)
                end) do
                    number
                else
                    0
                end
                solve(a, x, y2 + 1, result + current)
            end
        end
    end

    def solve(a) do
        solve(a, 0, 0, 0)
    end

    def to_array(line) do
        line
        |> String.to_charlist
        |> Enum.with_index
        |> Enum.map(fn({char, index}) -> {index, char} end)
        |> Map.new
    end

    def main() do
        input = IO.read(:stdio, :all)
            |> String.split("\n")
            |> Enum.with_index
            |> Enum.map(fn({line, index}) -> {index, to_array("." <> line <> ".")} end)
            |> Map.new
        IO.inspect(solve(input))
    end
end

Solution.main()