给我5分钟,带你秒杀所有图算法之DFS、BFS一、前言二

你好,我是小黄,一名独角兽企业的Java开发工程师。
感谢茫茫人海中我们能够相遇,
俗话说:当你的才华和能力,不足以支撑你的梦想的时候,请静下心来学习,
希望优秀的你可以和我一起学习,一起努力,实现属于自己的梦想。

一、前言

上一篇文章我们提到了关于图的形象化描述方法,不知道大家还有没有印象。没有印象的话,可以去看一下上期的内容

对于图来说,搜索的方法无外乎两种,深度优先搜索(DFS)和广度优先搜索(BFS)

两种搜索算法也不太相同,今天我们就来看一下这两个搜索算法

二、深度优先搜索

我们一提到深度优先搜索,脑子里第一时间想到的就是递归

没错,深搜就是依靠递归的方法来进行的搜索,我们来看一个例题:
在这里插入图片描述
对于上图来说,使用深度优先搜索的路线为:0 -> 3 - > 2 -> 4 -> 5 -> 1

这里不懂深搜的小伙伴可以看下这篇:深度优先搜索

递归版本:

	/**
     * 深度优先搜索
     * 
     * @param node
     * @param set
     */
	public void DFS(Node node, Set<Node> set) {
        if (node == null) {
            return;
        }
        if (!set.contains(node)) {
            set.add(node);
            System.out.print(node.value + " ");
            for (Node node1 : node.nexts) {
                DFS(node1, set);
            }
        }
    }
复制代码

迭代版本:

	/**
     * 深度优先搜索
     *
     * @param node
     */
	public void DFS(Node node) {
        Stack<Node> stack = new Stack<>();
        Set<Node> set = new HashSet<>();
        stack.add(node);
        set.add(node);
        System.out.print(node.value + " ");
        while (!stack.isEmpty()) {
            Node cur = stack.pop();
            for (Node next : cur.nexts) {
                if (!set.contains(next)) {
                    stack.add(cur); // 用来做记忆化的
                    stack.add(next);
                    System.out.print(next.value + " ");
                    set.add(next);
                    break;
                }
            }
        }
    }
复制代码

测试结果:

迭代版本:
0 3 2 4 5 1 
递归版本:
0 3 2 4 5 1 
复制代码

三、广度优先搜索

对于广度优先搜索的话,简单的来说,像走地图一样,一圈一圈的扩展开来

我们来看一个例题:
在这里插入图片描述
对于上图来说,使用深度优先搜索的路线为:0 -> 3 -> 1 -> 2 -> 4 -> 5

这里不懂广搜的小伙伴可以看下这篇:广度优先搜索

	/**
     * 广度优先搜索
     *
     * @param node
     */
    public static void BFS(Node node) {
        if (node == null) {
            return;
        }
        Queue<Node> queue = new LinkedList<>();
        // 代表是否被使用
        Set<Node> set = new HashSet<>();
        queue.add(node);
        set.add(node);
        while (!queue.isEmpty()) {
            Node cur = queue.poll();
            System.out.print(cur.value + " ");
            for (Node next : cur.nexts) {
                if (!set.contains(next)) {
                    queue.add(next);
                    set.add(next);
                }
            }
        }
    }
复制代码

四、结语

这期的深度优先搜索和广度优先搜索比较简单

让你对图的搜索大概有个了解,下几期将会讲解一些真实的算法

在算法题中,题目不会单纯的让你求深搜和广搜,经常会和别的一起出现,比如最小生成树等

本期的内容就到这里,算法源码可以在微信公众号回复:算法源码,即可获得链接,下期将会讲述 拓扑排序 算法

我是一名独角兽企业的Java开发工程师,希望可以点个关注呀,有问题可以留言或者私信加我微信,我们下期再见!