【LeetCode】210. Course Schedule II (Medium)

最新推荐文章于 2025-07-29 14:24:32 发布

原创最新推荐文章于 2025-07-29 14:24:32 发布 · 258 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索（DFS）进行拓扑排序的方法，适用于有向无环图（DAG）。通过构建图并追踪节点的发现时间和完成时间，可以有效地检测环并输出节点的拓扑顺序。

【题意】给出一个有向图的边集，输出拓扑排序

【解】根据边集建图，用DFS，找出每个节点的发现时间和完成时间，按完成时间从大到小输出

写的时候把有向边方向写反了。。。

class Solution {
public:
    vector<int> findOrder(int numCourses, vector<pair<int, int>>& prerequisites) {
        init(numCourses, prerequisites);
		int n = 0;
		for (int i = 0; i < numCourses; i++)
		{
			if (!visited[i])
			{
				visited[i] = true;
				pre[i] = n;
				n++;
				dfs(n, numCourses, i);
			}
		}
		if (!find)
		    result.resize(0);
		return result;
    }
    private:
	vector<vector<int>> graph;
	vector<bool> visited;
	vector<int> pre;
	vector<int> post;
	vector<int> result;
	stack<int> temp;
	bool find;
	void init(int numCourses, vector<pair<int, int>>& edges)
	{
		find = true;
		graph.resize(numCourses);
		int n = edges.size();
		for (int i = 0; i < n; i++)
			graph[edges[i].first].push_back(edges[i].second);
		visited.resize(numCourses);
		for (int i = 0; i < numCourses; i++)
			visited[i] = false;
		pre.resize(numCourses);
		for (int i = 0; i < numCourses; i++)
			pre[i] = -1;
		post.resize(numCourses);
		for (int i = 0; i < numCourses; i++)
			post[i] = -1;
	}
	void dfs(int& cur, int n, int start)
	{
		for (size_t i = 0; i < graph[start].size(); i++)
		{
			if (!visited[graph[start][i]])
			{
				visited[graph[start][i]] = true;
				pre[graph[start][i]] = cur;
				cur++;
				dfs(cur, n, graph[start][i]);
			}
			else //判断是不是DAG
			{
				if (pre[graph[start][i]] < pre[start] && post[graph[start][i]] == -1)
					find = false;
			}
		}
		post[start] = cur;
		cur++;
		result.push_back(start); //图建反了，push_back就可以了。。。
	}
};