En fin måte å kode dette på er å helle på traversering levert av Data.Foldable.
{-# LANGUAGE DeriveFunctor, DeriveFoldable #-}
import Data.Foldable
import Data.Monoid
Vi kan utlede en forekomst av det automatisk ved hjelp av en utvidelse, men vi trenger å endre rekkefølgen på feltene i Node konstruktøren å gi oss en in-order traversering.
Mens vi er i gang, skal vi fjerne begrensninger på datatype selv. De faktisk gir ingen nytte, og har blitt fjernet fra språket som for Haskell 2011. (Når du ønsker å bruke slike begrensninger bør du sette dem på instanser av klasser, ikke på datatype.)
data BST a
= Void
| Node
{ left :: BST a
, val :: a
, right :: BST a
} deriving (Eq, Ord, Read, Show, Foldable)
Først definerer vi hva det betyr for en liste for å være strengt sortert.
sorted :: Ord a => [a] -> Bool
sorted [] = True
sorted [x] = True
sorted (x:xs) = x < head xs && sorted xs
-- head is safe because of the preceeding match.
Da kan vi bruke toListmetoden levert av Data.Foldable, og de ovennevnte hjelper.
isBST :: Ord a => BST a -> Bool
isBST = sorted . toList
Vi kan også gjennomføre dette mer direkte, som du spurte. Siden vi fjernet uønskede begrensninger på datatypen, kan vi forenkle definisjonen av en fold.
cata :: (b -> a -> b -> b) -> b -> BST a -> b
cata _ z Void = z
cata f z (Node l x r) = f (cata f z l) x (cata f z r)
Nå trenger vi en datatype for å modellere et resultat av vår catamorphism, som er at vi har heller ingen noder ( Z), eller en rekke strengt økende noder ( T) eller har sviktet ( X)
data T a = Z | T a a | X deriving Eq
Og vi kan da gjennomføre isBSTdirekte
isBST' :: Ord a => BST a -> Bool
isBST' b = cata phi Z b /= X where
phi X _ _ = X
phi _ _ X = X
phi Z a Z = T a a
phi Z a (T b c) = if a < b then T a c else X
phi (T a b) c Z = if b < c then T a c else X
phi (T a b) c (T d e) = if b < c && c < d then T a e else X
Dette er litt kjedelig, så kanskje det ville være bedre å dekomponere måten vi komponere den midlertidige tilstander litt:
cons :: Ord a => a -> T a -> T a
cons _ X = X
cons a Z = T a a
cons a (T b c) = if a < b then T a c else X
instance Ord a => Monoid (T a) where
mempty = Z
Z `mappend` a = a
a `mappend` Z = a
X `mappend` _ = X
_ `mappend` X = X
T a b `mappend` T c d = if b < c then T a d else X
isBST'' :: Ord a => BST a -> Bool
isBST'' b = cata phi Z b /= X where
phi l a r = l `mappend` cons a r
Personlig ville jeg nok bare bruke Sammenleggbar eksempel.