本文主要是介绍课程表2问题,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
class Solution {// 存储有向图List<List<Integer>> edges;// 标记每个节点的状态:0=未搜索,1=搜索中,2=已完成int[] visited;// 用数组来模拟栈,下标 n-1 为栈底,0 为栈顶int[] result;// 判断有向图中是否有环boolean valid = true;// 栈下标int index;public int[] findOrder(int numCourses, int[][] prerequisites) {edges = new ArrayList<List<Integer>>();for (int i = 0; i < numCourses; ++i) {edges.add(new ArrayList<Integer>());}visited = new int[numCourses];result = new int[numCourses];index = numCourses - 1;for (int[] info : prerequisites) {edges.get(info[1]).add(info[0]);}// 每次挑选一个「未搜索」的节点,开始进行深度优先搜索for (int i = 0; i < numCourses && valid; ++i) {if (visited[i] == 0) {dfs(i);}}if (!valid) {return new int[0];}// 如果没有环,那么就有拓扑排序return result;}public void dfs(int u) {// 将节点标记为「搜索中」visited[u] = 1;// 搜索其相邻节点// 只要发现有环,立刻停止搜索for (int v: edges.get(u)) {// 如果「未搜索」那么搜索相邻节点if (visited[v] == 0) {dfs(v);if (!valid) {return;}}// 如果「搜索中」说明找到了环else if (visited[v] == 1) {valid = false;return;}}// 将节点标记为「已完成」visited[u] = 2;// 将节点入栈result[index--] = u;}
}
这篇关于课程表2问题的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!