algorithm - Haskell:如何更改算法以处理任何大小的列表？

Question

我有这个代码:

project= [ 
  [(a,b),(c,d),(e,f)] 
  | 
    a<-[1..5],
    b<-[1..3],
    c<-[1..5],
    d<-[1..3],
    e<-[1..5],
    f<-[1..3]

  , a*b + c*d + e*f <6
  , a + c + e == 5
  , b == 3 || d==3 || f==3
  ]

x=take 1 project
main = print $  x

它返回 3 对 [(x,y),(x,y),(x,y)] 的列表。有3个条件:

• 如果对所有 x 求和，则必须得到 5。
• 如果您对所有 x*y 求和，您将得到少于 6。
• 至少有一个y等于3。

现在，我希望完全相同的算法适用于任何更长的列表，例如 10 对。我应该怎么做？

Answer 1

这里:

project n =
  [ x
  | x <- replicateM n $ liftA2 (,) [1..5] [1..3]
  , sum (map (uncurry (*)) x) < 6
  , sum (map fst x) == 5
  , any ((==3) . snd) x
  ]
main = print $ take 1 $ project 3

或者像这样:

project n
  = filter (any ((==3) . snd))
  $ filter ((==5) . sum . map fst)
  $ filter ((<6) . sum . map (uncurry (*)))
  $ replicateM n
  $ liftA2 (,) [1..5] [1..3]