[亚马逊面试OA2] Order Dependency 排课

图论里面的基础问题。
有一堆课程,每个课程都有先后,比如:要上B,就必须先学完A,要上C,又必须先学完B。

例子.png

图中的例子:
0是1,3,4的先驱,4是1,5的先驱,1和3是2的先驱
这样的话,一个学生要上完所有的课,可能的顺序就是:[0,4,1,3,5,2]。这只是一种可能,只要能够上完,就算是正确。
给定了两个类:

    public class Course {
        String course = "";
        public Course(String str) {
            course = str;
        }
    }

    public class Dependency {
        Course cur;
        Course pre;
        public Dependency(Course cur, Course pre) {
            this.cur = cur;
            this.pre = pre;
        }
    }

输入是Dependency的数组,输出要求是Course的数组。
函数的signature长这样: public List<Course> find(List<Dependency> list)


这种题,不会做真就不会做。要是我不知道什么叫拓扑排序(Topological Sort),根本就是递归懵逼。
这道题就是典型的拓扑排序
拓扑排序就是拿来干这种事的,只要有先后顺序的,想想它准没错。

停!STOP!

怎么就直接说拓扑排序了呢?大家都没有发现,万一这个图中有环该怎么办?
在leetcode里面也有,就合着一块写了,不想看的就往下拉,题目的分析在下面~


问题就来了,如何检测有向图中是否有环
BFS?DFS?反正图,总跑不了这两个。想想我们人是怎么看有没有环的,不就是沿着每条路看有没有回头的?DFS就是它了!
这个算法叫什么着色来着,算法导论里面也写得很详细易懂。

有环.png

上面图就有环

这里用的算法叫做三色法。
当遍历到某一个节点时,节点可能会出现三种状态,这三种可能的状态用颜色来标记,便于理解。
首先定义一下三色(也就是三个状态):

  1. 白色:代表此节点从来没有被访问过
  2. 灰色:代表此节点在当前DFS遍历中被访问过
  3. 黑色:代表此节点在之前的DFS遍历中被访问过 且 从此节点出发的DFS也走完了的
算法的基本思想就是:

先在图中选出白色的节点,DFS遍历:

  • 如果发现有白色的节点就继续往下走,并标记为灰色
  • 如果发现遇到黑色的节点,就不在黑色当前节点处DFS了
  • 如果发现遇到灰色的节点,说明在当前DFS遍历中,这个节点之前被访问过,现在又被访问了,那么就有环了。
举个例子

上图为例。

  1. 先选择2节点开始DFS,标记2为灰色。
  2. 走到4,发现4是白色(没有被访问过),标记为灰色
    2.1: 发现4没有邻居了,返回后把4标记为黑色
  3. 回到2,再访问3,标记3为灰色
    3.1 :3节点可以继续往下走,走到1,发现1也是白色,标记为灰色
    3.2 :从1节点往下走,发现2是1的邻居,但是2是灰色的,说明有环,返回!

如果图里没有环,最终所有的节点都会被标记成黑色。(当然,island这种东西不考虑)

上代码

LeetCode: [207. Course Schedule]

public class Solution {
    public boolean canFinish(int numCourses, int[][] prerequisites) {
        if (numCourses <= 1 || prerequisites == null || prerequisites.length == 0) {
            return true;
        }
        ArrayList<Integer>[] graph = new ArrayList[numCourses];
        // 把int[][]转换成adjacency list
        // 在graph中,index是每一个课程
        // 所连接的arraylist是这个课程所相连的其他所有课程
        for (int i = 0; i < prerequisites.length; i++) {
            if (graph[prerequisites[i][0]] == null) {
                graph[prerequisites[i][0]] = new ArrayList<>();
            }
            graph[prerequisites[i][0]].add(prerequisites[i][1]);
        }
        // mark每个cell中具有3个值,0,1,2
        // 0 代表 没有被访问过
        // 1 代表 正在被访问
        int[] mark = new int[numCourses];
        for (int i = 0; i < numCourses; i++) {
            if (!dfs(mark, graph, i)) {
                return false;
            }
        }

        return true;
    }
    
    private boolean dfs(int[] mark, ArrayList[] graph, int course) {
        if (mark[course] == 1) {
            return false;
        } else if (mark[course] == 2) {
            return true;
        } else {
            mark[course] = 1; // 现在正在访问中
        }

        // 继续访问该课程相连的所有课程
        if (graph[course] != null) {
            for (int i = 0; i < graph[course].size(); i++) {
                if (!dfs(mark, graph, (int)graph[course].get(i))) {
                    return false;
                }
            }
        }
        mark[course] = 2;
        return true;
    }
}

这方法是比较经典,也比较快的方法,leetcode里面击败了来自全球的96%的Java代码

好,下面进入正题


拓扑排序

数据结构:
  1. 图的表示:Adjacency List
  1. 一个入度记录表:HashMap
  2. BFS必备数据结构:Queue
算法思路:

入度代表的是这个节点被指向的次数。在例子那张图中,1节点的入度为2, 0节点的入度为0,以此类推。
找入度为0的节点,BFS找它的所有邻居,然后把找到的所有邻居在入度记录表里面的入度数减一,如果某个邻居节点的入度数变成了0,那么加入Queue里面,成为下一个BFS的起点。

其实说白了,就是先挑没有prerequisite的课上,上完之后看有没有解锁的课,如果有,那么接着上,如果没有,看看有没有其他的课没有prerequisite。就像游戏里面什么修仙练级之类的……

上代码
public class OrderDependency {
    public static class Order {
        String order = "";
        public Order(String str) {
            order = str;
        }
    }

    public static class Dependency {
        Order cur;
        Order pre;
        public Dependency(Order cur, Order pre) {
            this.cur = cur;
            this.pre = pre;
        }
    }

    public static List<Order> find(List<Dependency> list) {
        List<Order> result = new ArrayList<>();
        if (list == null || list.size() == 0) {
            return result;
        }
        int totalOrder = 0;
        Map<Order, List<Order>> adjList = new HashMap<>();
        Map<Order, Integer> indegree = new HashMap<>();
        Queue<Order> queue = new LinkedList<>();

        // fill indegree and adjList
        for (Dependency dep : list) {
            Order cur = dep.cur;
            Order pre = dep.pre;
            // add to adjList
            if (adjList.containsKey(pre)) {
                adjList.get(pre).add(cur);
            } else {
                adjList.put(pre, new ArrayList<>());
                adjList.get(pre).add(cur);
            }
            // add to indegree
            if (indegree.containsKey(cur)) {
                indegree.put(cur, indegree.get(cur) + 1);
            } else {
                indegree.put(cur, 1);
            }
            if (!indegree.containsKey(pre)) {
                indegree.put(pre, 0);
            }
        }

        for (Order o : indegree.keySet()) {
            if (indegree.get(o) == 0) {
                queue.offer(o);
            }
        }

        // BFS
        while (!queue.isEmpty()) {
            Order order = queue.poll();
            indegree.remove(order);
            result.add(order);
            if (adjList.containsKey(order)) {
                for (Order cur : adjList.get(order)) {
                    if (indegree.containsKey(cur)) {
                        indegree.put(cur, indegree.get(cur) - 1);
                    }
                    if (indegree.get(cur) == 0) {
                        queue.offer(cur);
                    }
                }
            }
        }

        if (indegree.size() != 0) {
            return null;
        }
        return result;
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,542评论 6 504
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,822评论 3 394
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,912评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,449评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,500评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,370评论 1 302
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,193评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,074评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,505评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,722评论 3 335
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,841评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,569评论 5 345
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,168评论 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,783评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,918评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,962评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,781评论 2 354

推荐阅读更多精彩内容