瀏覽代碼

include initial verticies in solution list

pull/16/head
Tristan Sloughter 10 年之前
父節點
當前提交
eb5c0eb424
共有 1 個檔案被更改,包括 4 行新增3 行删除
  1. +4
    -3
      src/rebar_digraph.erl

+ 4
- 3
src/rebar_digraph.erl 查看文件

@ -18,7 +18,9 @@ restore_graph({Vs, Es}) ->
Graph.
solve(Graph, Vertices) ->
solve(Graph, Vertices, dict:new()).
solve(Graph, Vertices, lists:foldl(fun({Key, _}=N, Solution) ->
dict:store(Key, N, Solution)
end, dict:new(), Vertices)).
solve(_Graph, [], Solution) ->
{_, Vertices} = lists:unzip(dict:to_list(Solution)),
@ -28,8 +30,7 @@ solve(Graph, Vertices, Solution) ->
lists:foldl(fun(V, {NewVertices, SolutionAcc}) ->
OutNeighbors = digraph:out_neighbours(Graph, V),
lists:foldl(fun({Key, _}=N, {NewVertices1, SolutionAcc1}) ->
case dict:is_key(Key, SolutionAcc1) orelse
lists:keymember(Key, 1, Vertices) of
case dict:is_key(Key, SolutionAcc1) of
true ->
{NewVertices1, SolutionAcc1};
false ->

Loading…
取消
儲存