首页 > 解决方案 > 带有两个参数的 Haskell 列表过滤

问题描述

我必须编写一个函数,该函数使用一个产生的参数True进行过滤,然后使用另一个产生的参数进行过滤False

我试过这个:

selectUnless :: (t -> Bool) -> (t -> Bool) -> [t] -> [t]
selectUnless fx gx (x:xs) = filter gx (filter fx (x:xs))

但我需要那个“不是 gx”的列表。

例如:

selectUnless (>= 2) (==2) [1,2,3,4] == [3,4]
selectUnless even odd [1..50] == [2,4..50]

标签: listhaskellfilter

解决方案


Haskell 有一个not :: Bool -> Bool转换True成的运算符,False反之亦然。

问题当然不是gx一个,而是一个函数。然而,我们可以构造一个函数,如:Boolt -> Bool

\x -> not (gx x)

因此,这将适用not于 的结果gx x。或者我们可以使用(.) :: (b -> c) -> (a -> b) -> a -> c,例如:

selectUnless :: (t -> Bool) -> (t -> Bool) -> [t] -> [t]
selectUnless fx gx = filter (not . gx) . filter fx

我们甚至可以使用liftA2 :: Applicative f => (a -> b -> c) -> f a -> f b -> f c

import Control.Applicative(liftA2)

selectUnless :: (t -> Bool) -> (t -> Bool) -> [t] -> [t]
selectUnless = (filter .) . (. (not .)) . liftA2 (&&)

或者@JonPurdy提供的更优雅的解决方案:

import Control.Applicative(liftA2)
import Data.Function(on)

pre :: (a -> b) -> (b -> c) -> a -> c
pre = flip (.)

filtered :: (([a] -> [a]) -> ([a] -> [a]) -> c) -> (a -> Bool) -> (a -> Bool) -> c
filtered = (`on` filter)

selectUnless :: (a -> Bool) -> (a -> Bool) -> [a] -> [a]
selectUnless = pre (not .) . filtered (.)

请注意,您的原始代码无法处理空列表:实际上,您只(x:xs)在函数中为列表指定了模式,而不是[]. 然而,这里没有必要使用额外的模式,filter已经可以处理空列表和非空列表。


推荐阅读