Haskell红黑树:如何从工作表批量插入多值至树结构?
解决Haskell红黑树批量插入的问题
你的代码核心问题有两个:
- 用
map处理列表时,每个元素都插入到初始树t,最终得到的是多个独立的树(类型为[Tree a]),而非累积更新后的单个树; - 类型签名错误,
[b]应改为[a]——insert要求元素类型与树的元素类型一致,且需要Ord约束。
要实现基于最新树状态的批量插入,应该用**折叠(fold)**来累积状态,foldl会遍历列表,每次将前一次插入后的树作为下一次的输入。
简洁实现
add_elements :: Ord a => Tree a -> [a] -> Tree a add_elements = foldl (flip insert)
展开式(更易理解)
如果觉得简洁写法太抽象,可展开为显式的累积函数:
add_elements initialTree elements = foldl accumulate initialTree elements where accumulate currentTree x = insert x currentTree
辅助函数(从空树构建)
若经常需要从空树开始插入列表元素,可定义:
fromList :: Ord a => [a] -> Tree a fromList = add_elements Leaf
用法示例:fromList [5,2,7,1] 会生成包含这些元素的红黑树。
为什么不用map?
map ins list会对每个x执行insert x t(t是最初传入的树),所有操作相互独立,不会保留之前插入的结果。而foldl会把每次插入后的新树传递给下一次操作,实现状态的逐步累积。
内容的提问来源于stack exchange,提问作者student422
相关产品推荐
相关产品推荐

