Sha256: 665b69079327c931201c61f3417e59e76d0e2a486c2ab6b6ebca672204678469
Contents?: true
Size: 1.55 KB
Versions: 69
Compression:
Stored size: 1.55 KB
Contents
defmodule Triplet do
@doc """
Calculates sum of a given triplet of integers.
"""
@spec sum([non_neg_integer]) :: non_neg_integer
def sum(triplet) do
Enum.reduce(triplet, 0, &(&1 + &2))
end
@doc """
Calculates product of a given triplet of integers.
"""
@spec product([non_neg_integer]) :: non_neg_integer
def product(triplet) do
Enum.reduce(triplet, 1, &(&1 * &2))
end
@doc """
Determines if a given triplet is pythagorean. That is, do the squares of a and b add up to the square of c?
"""
@spec pythagorean?([non_neg_integer]) :: boolean
def pythagorean?([a, b, c]) do
a * a + b * b == c * c
end
defp select?(triplet) do
pythagorean?(triplet)
end
defp select?(triplet, sum) do
pythagorean?(triplet) && sum(triplet) == sum
end
@doc """
Generates a list of pythagorean triplets from a given min to a given max.
"""
@spec generate(non_neg_integer, non_neg_integer) :: [list(non_neg_integer)]
def generate(min, max) do
for x <- Enum.to_list(min..max),
y <- Enum.to_list(x..max),
z <- Enum.to_list(y..max),
select?([x, y, z]),
do: [x, y, z]
end
@doc """
Generates a list of pythagorean triplets from a given min to a given max, whose values add up to a given sum.
"""
@spec generate(non_neg_integer, non_neg_integer, non_neg_integer) :: [list(non_neg_integer)]
def generate(min, max, sum) do
for x <- Enum.to_list(min..max),
y <- Enum.to_list(x..max),
z <- Enum.to_list(y..max),
select?([x, y, z], sum),
do: [x, y, z]
end
end
Version data entries
69 entries across 69 versions & 1 rubygems