Haskell:根据成员类型设置过滤器？

Question

假设我在 Haskell 中有以下数据结构来表示跳棋/草稿:<​​/p>

data Piece = Reg {pos :: Square, color :: Color}
         | King {pos :: Square, color :: Color}
    deriving (Show, Eq)

给定这些棋子的列表，我如何从列表中分离出K？我一直在查看 Data.Set 的文档 http://www.haskell.org/ghc/docs/7.6.2/html/libraries/containers-0.5.0.0/Data-Set.html但找不到对我来说似乎显而易见的东西。

简而言之，我需要一种方法，在给定 Data.Set 组 Piece 的情况下，返回所有 King 类型的子集件。我觉得这是非常简单的事情，但我还没有遇到过，因为我是 Haskell 中的 Data.Set 类的新手。

Answer 1

您可以定义一个 bool 函数isKing，然后在Data.Set中使用filter，如下:

import Data.Set as S
data Color = Int deriving (Show, Eq)
data Square = Square (Int,Int) deriving (Show, Eq)
data Piece = Reg {pos :: Square, color :: Color}
         | King {pos :: Square, color :: Color}
    deriving (Show, Eq)

isKing King{} = True
isKing _ = False

getKings s = S.filter isKing s