functional-programming - 通过查看第二个列表中的 bool 值来过滤一个列表的元素

Question

我有两个等长的列表。我想通过查看第二个列表中具有相同索引的元素是否具有真 bool 值来过滤第一个列表的元素。

Example:

[1,2,3,4,5]:int list

[true,false,false,true,false]:bool list

Expected result: [1,4]

我知道有两种方法可以实现这一点:

1) 编写一个接受两个列表的函数。对于第一个列表中我要追加的每个元素，检查第二个列表的当前(头)元素是否为真。

2) 压缩两个列表并根据 bool 值过滤。

应该有更简单的方法，对吧？

Answer 1

不是真的。最干净的方法可能是

List.map (fn (x,y) => x) (List.filter (fn (x,y) => y) (ListPair.zip (L1,L2)))

或

List.map Option.valOf (List.filter Option.isSome (ListPair.map(fn (x,y) => if y then SOME x else NONE) (L1,L2)))

递归函数也不错:

fun foo ([],[]) = []
  | foo ([],L) = raise Fail "Different lengths"
  | foo (L,[]) = raise Fail "Different lengths"
  | foo (x::xs, b::bs) = if b then x::foo(xs,bs) else foo(xs,bs)