LintCode: 拓扑排序

描述

给定一个有向图，图节点的拓扑排序被定义为：

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

找到给定图的任一拓扑排序

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

说明

Learn more about representation of graphs

样例

对于下列图：

这个图的拓扑排序可能是：

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

或者

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

或者

....

思路：

图的DFS，BFS可以看这儿：https://www.jianshu.com/p/70952b51f0c8

拓扑排序： 每次将入度为0的节点保存，再从图中删掉，更新剩余节点入度，依次迭代，就可以了。

代码：

public ArrayList<DirectedGraphNode> topSort(ArrayList<DirectedGraphNode> graph) {
        // write your code here
        ArrayList<DirectedGraphNode> result = new ArrayList<>();
        HashMap<DirectedGraphNode, Integer> map = new HashMap<>();
        for (DirectedGraphNode node:graph){
            for(DirectedGraphNode ner:node.neighbors){
                if(map.containsKey(ner)){
                    map.put(ner,map.get(ner)+1);
                }else {
                    map.put(ner,1);
                }
            }
        }
        Queue<DirectedGraphNode> q = new LinkedList<>();
        for(DirectedGraphNode node : graph){
            if(!map.containsKey(node)){
                q.offer(node);
                result.add(node);
            }
        }
        while (!q.isEmpty() ){
            DirectedGraphNode node = q.poll();
            for (DirectedGraphNode ner:node.neighbors){
                if(map.get(ner)==1){
                    q.offer(ner);
                    result.add(ner);
                    map.remove(ner);
                }else {
                    map.put(ner, map.get(ner)-1);
                }
            }
        }
        return result;
    }

DFS也可以实现： 

见 : https://www.jiuzhang.com/solution/topological-sorting/#tag-other