haskell - Haskell 中使用无穷列表的毕达哥拉斯三元组

Question

我想使用无穷列表在 Haskell 中生成毕达哥拉斯三元组。我的代码有什么问题:

trojkaty = [(a,b,c) | a <- [1..], b <- [1..], c <- [1..], (a^2)+(b^2)==(c^2)]

Answer 1

尝试用c的中间值来表达a和b的上限，否则会强制整个列表在检查最后一个条件之前的无限值。

trojkaty :: [(Int, Int, Int)]
trojkaty = [(a,b,c) | c <- [2..], b <- [2..c-1], a <- [2..b-1], a^2 + b^2 == c^2]

main = do
  print $ take 5 trojkaty