127. 拓扑排序

127. 拓扑排序

 

给定一个有向图，图节点的拓扑排序定义如下:

• 对于图中的每一条有向边  A -> B  , 在拓扑排序中A一定在B之前.
• 拓扑排序中的第一个节点可以是图中的任何一个没有其他节点指向它的节点.

针对给定的有向图找到任意一种拓扑排序的顺序.

样例

For graph as follow:

The topological order can be:

[0, 1, 2, 3, 4, 5]

[0, 2, 3, 1, 5, 4]

...

挑战

能否分别用BFS和DFS完成？

说明

有关图的表示详情请看这里

注意事项

你可以假设图中至少存在一种拓扑排序

 

 

public ArrayList<DirectedGraphNode> topSort(ArrayList<DirectedGraphNode> graph) {

    ArrayList<DirectedGraphNode> results = new ArrayList<>();

    HashMap<DirectedGraphNode, Integer> map = new HashMap<>();

    for (DirectedGraphNode node : graph) {

        for (DirectedGraphNode neighbor : node.neighbors) {

            if (map.containsKey(neighbor)) {

                map.put(neighbor, map.get(neighbor) + 1);

            } else {

                map.put(neighbor, 1);

            }

        }

    }

 

 

    Queue<DirectedGraphNode> queue = new LinkedList<>();

    for (DirectedGraphNode node : graph) {

        if (!map.containsKey(node)) {

            queue.offer(node);

            results.add(node);

        }

    }

 

 

    while (!queue.isEmpty()) {

        DirectedGraphNode node = queue.poll();

        for (DirectedGraphNode neighbor : node.neighbors) {

            map.put(neighbor, map.get(neighbor) - 1);

            if (map.get(neighbor) == 0) {

                results.add(neighbor);

                queue.offer(neighbor);

            }

        }

    }

    return results;

}