fix bug in divisors code

2025-03-01 21:47:42 -05:00 · 2025-03-01 21:47:42 -05:00 · ba075f979f
commit ba075f979f
parent 934474326a
1 changed files with 3 additions and 2 deletions
--- a/problems/21/solution.exs
+++ b/problems/21/solution.exs
@ -2,8 +2,9 @@ defmodule AmicableNumbers do
  def divisors(n) do
    [ 1 |
    (for i <- 2..floor(:math.sqrt(n))//1,
-      div(n, i) == n / i do
-        [i, div(n, i)]
+      j = div(n, i),
+      j == n / i do
+        if i !== j, do: [i,j], else: [i]
    end |> Enum.flat_map(&(&1))) ]
  end
 end