Skip to main content

Haskell: Breadth-first Tree Traversal

As an exercise, I implemented breadth-first tree traversal (instead of the much more common depth-first tree traversal) in Haskell. It took me about 3 hours, mainly because I haven't even finished reading the turorial yet. It really struck me how truly lazy Haskell is! My own language, Squeamish, had lazy evaluation, but in Haskell, you have to somewhat bend over backwards with the imperative stuff to say, "Yeah, really, I'm done defining stuff. Uh, can you actually, uh, do some evaluation or something, and, uh, maybe even print the answer?"
{- Implement breadth-first tree traversal.

Name: Shannon -jj Behrens
Date: Tue Dec 13 03:18:34 PST 2005 -}

module Main where

-- The main definition of a tree.
data Tree a = Leaf a | Branch a (Tree a) (Tree a)

-- Depth-first tree traversal.
depthFirst :: Tree a -> [a]
depthFirst (Leaf x) = [x]
depthFirst (Branch x left right) = depthFirst left ++ [x] ++
depthFirst right

-- Breadth-first tree traversal.
breadthFirst :: Tree a -> [a]
breadthFirst x = _breadthFirst [x]

_breadthFirst :: [Tree a] -> [a]
_breadthFirst [] = []
_breadthFirst xs = map treeValue xs ++
_breadthFirst (concat (map immediateChildren xs))

-- Get the value of a given tree.
treeValue :: Tree a -> a
treeValue (Leaf x) = x
treeValue (Branch x left right) = x

-- Get the immediate children of a tree.
immediateChildren :: Tree a -> [Tree a]
immediateChildren (Leaf x) = []
immediateChildren (Branch x left right) = [left, right]

-- Define some tree.
mytree = Branch "1"
(Branch "2"
(Leaf "4")
(Leaf "5"))
(Branch "3"
(Leaf "6")
(Leaf "7"))

{- 1
/ 2 3
/ \ / 4 5 6 7

Here's another.
mytree = Branch "0"
(Leaf "1")
(Branch "2"
(Branch "3"
(Leaf "4")
(Leaf "5"))
(Leaf "6"))

0
/ 1 2
/ 3 6
/ 4 5 -}

-- Create one "do" out of a list of things to do.
doList :: [IO ()] -> IO ()
doList = foldr (>>) (return ())

-- Main.
main :: IO ()
main = do doList (map putStrLn (breadthFirst mytree))
The result is:
$ runhugs main.hs
1
2
3
4
5
6
7

Comments

xerox said…
First of all, welcome to Haskell! :-)

I don't know if you're used to the IRC, but there is an active community on the #haskell channel on the irc.freenode.net network - you're very welcome to join!

Nice piece of code, buddy. I couldn't find a way to use the PRE or CODE or whatever tags in writing the comment, so, if you can, feel free to edit and fix it. Stupid blogger.com.

Before starting the trip, note that I just tried to write a more idiomatic (and well indented, at least to the most common taste) version of the code - for fun purposes mainly. Also, I want to show you some interesting properties emerging from your code, check them out.

Adding a 'deriving (..)' clause makes the datatype automatically instantiated in the given classes. Show and Read make values of the given type serializable via the functions 'read' and 'show'. Deriving Show is also useful for debugging purposes (if you happened to encounter any error of the type "No Show instance for (Tree a)" or something.)

data Tree a = Leaf a | Branch a (Tree a) (Tree a) deriving (Show,Read)

Right on!

depthFirst :: Tree a -> [a]
depthFirst (Leaf x) = [x]
depthFirst (Branch x l r) = depthFirst l ++ [x] ++ depthFirst r

Is it more common to hide the helper functions/values in a 'where' clause.

I don't know again how far are you on your tutorial, but keep this in mind: in the List Monad, return is \x -> [x], and bind is exactly concatMap: (>>=) = concatMap.

breadthFirst :: Tree a -> [a]
breadthFirst x = breadthFirst' [x]
where breadthFirst' :: [Tree a] -> [a]
breadthFirst' [] = []
breadthFirst' xs = map treeValue xs ++ breadthFirst' (xs >>= immediateChildren)

Pattern matching unused arguments is usually done by the means of the '_' notation. To please the eye.

treeValue :: Tree a -> a
treeValue (Leaf x) = x
treeValue (Branch x _ _) = x

immediateChildren :: Tree a -> [Tree a]
immediateChildren (Leaf _) = []
immediateChildren (Branch _ l r) = [l, r]

The next function is straightforwardly taken from the standard library,take a look around there: http://darcs.complete.org/fptools/libraries/base/Data/Tree.hs

levels = map (map treeValue) . takeWhile (not . null) . iterate (>>= immediateChildren) . return

Values of types instantiated in the 'Num a' typeclass (like Int, to say), along with many others (try :info Show in hugs or GHCi), are also instantiated 'Show a'; meaning that they can be feeded to the 'show' function, and by extension to 'print', latter.

Note that: print = putStrLn . show

mytree :: Tree Int
mytree = Branch 1
(Branch 2
(Leaf 4)
(Leaf 5))
(Branch 3
(Leaf 6)
(Leaf 7))

Your 'doLine' is just sequence_ from Control.Monad (the '_' means that the result of the actions in the list are discarded, take a look at the documentation: http://haskell.org/ghc/docs/latest/html/libraries/index.html).

But then, there is a simpler way:

main :: IO ()
main = mapM_ (print) (breadthFirst mytree)

Happy happy, joy joy.

-- xerox on #haskell
Anonymous said…
Dou have explanation for this ?
In OOP PHP implementation ?
jjinux said…
Sorry, no.

Popular posts from this blog

Ubuntu 20.04 on a 2015 15" MacBook Pro

I decided to give Ubuntu 20.04 a try on my 2015 15" MacBook Pro. I didn't actually install it; I just live booted from a USB thumb drive which was enough to try out everything I wanted. In summary, it's not perfect, and issues with my camera would prevent me from switching, but given the right hardware, I think it's a really viable option. The first thing I wanted to try was what would happen if I plugged in a non-HiDPI screen given that my laptop has a HiDPI screen. Without sub-pixel scaling, whatever scale rate I picked for one screen would apply to the other. However, once I turned on sub-pixel scaling, I was able to pick different scale rates for the internal and external displays. That looked ok. I tried plugging in and unplugging multiple times, and it didn't crash. I doubt it'd work with my Thunderbolt display at work, but it worked fine for my HDMI displays at home. I even plugged it into my TV, and it stuck to the 100% scaling I picked for the othe

Drawing Sierpinski's Triangle in Minecraft Using Python

In his keynote at PyCon, Eben Upton, the Executive Director of the Rasberry Pi Foundation, mentioned that not only has Minecraft been ported to the Rasberry Pi, but you can even control it with Python . Since four of my kids are avid Minecraft fans, I figured this might be a good time to teach them to program using Python. So I started yesterday with the goal of programming something cool for Minecraft and then showing it off at the San Francisco Python Meetup in the evening. The first problem that I faced was that I didn't have a Rasberry Pi. You can't hack Minecraft by just installing the Minecraft client. Speaking of which, I didn't have the Minecraft client installed either ;) My kids always play it on their Nexus 7s. I found an open source Minecraft server called Bukkit that "provides the means to extend the popular Minecraft multiplayer server." Then I found a plugin called RaspberryJuice that implements a subset of the Minecraft Pi modding API for B

Creating Windows 10 Boot Media for a Lenovo Thinkpad T410 Using Only a Mac and a Linux Machine

TL;DR: Giovanni and I struggled trying to get Windows 10 installed on the Lenovo Thinkpad T410. We struggled a lot trying to create the installation media because we only had a Mac and a Linux machine to work with. Everytime we tried to boot the USB thumb drive, it just showed us a blinking cursor. At the end, we finally realized that Windows 10 wasn't supported on this laptop :-/ I've heard that it took Thomas Edison 100 tries to figure out the right material to use as a lightbulb filament. Well, I'm no Thomas Edison, but I thought it might be noteworthy to document our attempts at getting it to boot off a USB thumb drive: Download the ISO. Attempt 1: Use Etcher. Etcher says it doesn't work for Windows. Attempt 2: Use Boot Camp Assistant. It doesn't have that feature anymore. Attempt 3: Use Disk Utility on a Mac. Erase a USB thumb drive: Format: ExFAT Scheme: GUID Partition Map Mount the ISO. Copy everything from