在Erlang中,如何编写BFS树搜索以查找特定节点

时间:2011-04-29 11:45:20

标签: erlang

我想编写一个函数来查找BFS树搜索中的特定节点。

如何在Erlang中完成?

1 个答案:

答案 0 :(得分:4)

假设结构简单:

node() :: {Key :: term(), Left :: node(), Right :: node()} | undefined.

此函数将对树执行广度优先搜索并返回找到的第一个元素的深度(如果未找到节点,则返回false):

find(_Key, undefined) -> false;
find(Key,  Tree)      -> find(Key, [Tree], [], 0).

find(_Key, [], [], _Depth) -> % No more children
    false;
find(Key, [], Children, Depth) -> % Next level, increase depth
    find(Key, Children, [], Depth + 1);
find(Key, [{Key, _, _}|_Rest], _Children, Depth) -> % Found it!
    Depth;
find(Key, [{_, Left, Right}|Rest], Children, Depth) -> % Add children to search
    find(Key, Rest, add(Right, add(Left, Children)), Depth).

add(undefined, List) -> List;
add(E,         List) -> [E|List].

(空树只是undefined)。