Я работаю над книгой Programming Elixir и играю с упражнениями. У меня есть функция суммы, в которой я хотел бы вернуть значение суммы в списке [], а не просто число, но я получаю bad argument in arithmetic expression. Не могли бы сделать с некоторыми направлениями?
defmodule Mymod do
def sum([]), do: []
def sum([head | tail]) do
[head + sum(tail)]
end
end
Проблема в том, что Mymod.sum/1 возвращает список, а вы пытаетесь добавить его к номеру.
Чтобы это работало, вы должны «раскрыть» список, возвращаемый sum, в число, а затем добавить его:
defmodule Mymod do
def sum([]), do: []
def sum([head | tail]) do
case sum(tail) do
[] -> [head]
[s] -> [head + s]
end
end
end
IO.inspect Mymod.sum([5, 6, 7])
#⇒ [18]
Вариант хвостовой рекурсии будет выглядеть так (вдохновленный ответом @theanh-le):
defmodule Mymod do
def sum(list, acc \\ 0)
def sum([], 0), do: []
def sum([], acc), do: [acc]
def sum([head | tail], acc) do
sum(tail, head + acc)
end
end
IO.inspect Mymod.sum([5, 6, 7])
#⇒ [18]
Есть способ, которым вы можете использовать tail recursive, чтобы сделать ваш код намного красивее:
defmodule MyMod do
def sum([]), do: []
def sum(list) do
do_sum(0, list)
end
defp do_sum(current_sum, []), do:[current_sum]
defp do_sum(current_sum, [head | tail]) do
new_sum = current_sum + head
do_sum(new_sum, tail)
end
end
икс:
iex()> MyMod.sum([5,6,7])
[18]
[0] для пустого списка, в то время как OP вернул пустой список. Скорее всего, потребуется дополнительный defp do_sum(0, []), do:[]. 19.09.2016 
