如何在Julia中实现二进制搜索树?

时间:2016-04-03 09:04:15

标签: julia

我正在尝试在Julia中实现BST,但是当我调用insert函数时遇到了问题。当我尝试创建新节点时,结构保持不变。

我的代码:

type Node
    key::Int64
    left
    right
end

function insert(key::Int64, node)
    if node == 0
        node = Node(key, 0, 0)
    elseif key < node.key
        insert(key, node.left)
    elseif key > node.key
        insert(key, node.right)
    end
end

root = Node(0,0,0)
insert(1,root)
insert(2,root)

我也尝试将零改为零。我尝试的下一个版本是Node中定义的数据类型,但是当我尝试调用insert时没有任何值(类似于C Null)时,它给了我错误。

感谢回答。

1 个答案:

答案 0 :(得分:16)

代码有很多问题。 一个是它不是类型稳定的,左和右可以包含任何东西。 另一个是在insert函数中设置局部变量节点不会影响任何变化。 一个风格问题,修改其参数的函数通常都有!作为名称中的最后一个字符,例如insert!,push! setindex!

我认为以下内容应该有效:

type BST
    key::Int
    left::Nullable{BST}
    right::Nullable{BST}
end

BST(key::Int) = BST(key, Nullable{BST}(), Nullable{BST}())
BST() = BST(0)

function Base.push!(node::BST, key)
    if key < node.key
        if node.left.isnull
            node.left = BST(key)
        else
            push!(node.left.value, key)
        end
    elseif key > node.key
        if node.right.isnull
            node.right = BST(key)
        else
            push!(node.right.value, key)
        end
    end
end

root = BST()
push!(root, 1)
push!(root, 2)

这个版本超载了朱莉娅推!函数,并使用Nullable类型,以便它可以区分叶节点和它需要继续搜索的位置,以找到插入密钥的位置。这是一个递归定义,并没有进行优化,用循环代替递归会快得多。