lintcode127. 拓扑排序

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

对于图中的每一条有向边 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]

https://media-cdn.jiuzhang.com/markdown/images/8/6/91cf07d2-b7ea-11e9-bb77-0242ac110002.jpg
挑战
能否分别用BFS和DFS完成？

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

/**
 * Definition for Directed graph.
 * struct DirectedGraphNode {
 *     int label;
 *     vector<DirectedGraphNode *> neighbors;
 *     DirectedGraphNode(int x) : label(x) {};
 * };
 */

class Solution {
public:
    /*
     * @param graph: A list of Directed graph node
     * @return: Any topological order for the given graph.
     */
    vector<DirectedGraphNode*> topSort(vector<DirectedGraphNode*>& graph) {
        // write your code here
        map<DirectedGraphNode*,int>count;
        vector<DirectedGraphNode*>res;
        for (int i = 0; i < graph.size(); i++) {
            /* code */
            for(int j = 0; j < graph[i]->neighbors.size(); j++)
            {
                count[graph[i]->neighbors[j]]++;
            }
        }
        queue<DirectedGraphNode*>q;
        for (int i = 0; i < graph.size(); i++) {
            /* code */
            if(count[graph[i]]==0)
            {
                q.push(graph[i]);
                res.push_back(graph[i]);
            }
        }
        while(!q.empty())
        {
            DirectedGraphNode*cur=q.front();
            q.pop();
            for (int i = 0; i < cur->neighbors.size(); i++) {
                /* code */
                count[cur->neighbors[i]]--;
                if(count[cur->neighbors[i]]==0) 
                {
                    q.push(cur->neighbors[i]);
                    res.push_back(cur->neighbors[i]);
                }
            }
            
        }
        return res;
    }
};