给我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开发工程师，希望可以点个关注呀，有问题可以留言或者私信加我微信，我们下期再见！