文章目录
- 不要摆,没事干就刷题,只有好处,没有坏处,实在不行,看看竞赛题
- 面试经典 150 题 - 2
- 210. 课程表 II
不要摆,没事干就刷题,只有好处,没有坏处,实在不行,看看竞赛题
面试经典 150 题 - 2
面试经典 150 题
210. 课程表 II
210. 课程表 II
- 一眼拓扑排序. 好久没写过拓扑排序了,写得特别糟糕
public int[] findOrder(int n, int[][] prerequisites) {int[] order = new int[n];if (prerequisites == null) {for (int i = 0; i < n; i++) order[i] = i;return order;}// 创建邻接表 和 入度数组ArrayList<ArrayList<Integer>> adj = new ArrayList<>();for (int i = 0; i < n; i++) {adj.add(new ArrayList<>());}int[] inDegree = new int[n];for (int[] prerequisite : prerequisites) {adj.get(prerequisite[1]).add(prerequisite[0]);inDegree[prerequisite[0]]++;}// 入度队列 (不需要栈)Stack<Integer> s = new Stack<>();for (int i = 0; i < inDegree.length; i++) {if (inDegree[i] == 0) s.push(i);}// 拓扑排序int cnt = 0;for (int i = 0; i < n; i++) {if (s.isEmpty()) break;Integer pop = s.pop();order[cnt++] = pop;for (Integer x : adj.get(pop)) {inDegree[x]--;if (inDegree[x] == 0) {s.push(x);}}}if (cnt < n) return new int[0];return order;
}
- 看了下大佬的做法,发现确实有几处值得修改
主要就是度为0的不必非要用栈,用队列也行,队列直接作为拓扑排序的终止条件即可
没有前置关系时不需要要特判,全是度为0的节点,也可以照常执行
不要用statck,继承了Vector, 有很多锁,效率很低
修改后4ms,差不多了吧
public int[] findOrder(int n, int[][] prerequisites) {// 创建邻接表 和 入度数组ArrayList<ArrayList<Integer>> adj = new ArrayList<>();for (int i = 0; i < n; i++) adj.add(new ArrayList<>());int[] inDegree = new int[n];for (int[] prerequisite : prerequisites) {adj.get(prerequisite[1]).add(prerequisite[0]);inDegree[prerequisite[0]]++;}// 入度队列 (不需要栈)Deque<Integer> q = new LinkedList<>();for (int i = 0; i < inDegree.length; i++) {if (inDegree[i] == 0) q.offer(i);}// 拓扑排序int[] order = new int[n];int cnt = 0;while (!q.isEmpty()){Integer pop = q.poll();order[cnt++] = pop;for (Integer x : adj.get(pop)) {inDegree[x]--;if (inDegree[x] == 0) q.push(x);}}if (cnt < n) return new int[0];return order;
}