深度优先搜索实例

最新推荐文章于 2024-08-03 11:35:54 发布

wyc_

最新推荐文章于 2024-08-03 11:35:54 发布

阅读量3.2k

点赞数

分类专栏：算法与数据结构

本文链接：https://blog.csdn.net/cumtwyc/article/details/41957229

版权

本文通过11个实例深入探讨深度优先搜索（DFS）在验证二叉搜索树、恢复二叉搜索树、求最大深度、从先序和中序序列构造树等经典问题中的应用。每个问题都提供了相应的解决方案和代码实现。

摘要由CSDN通过智能技术生成

前面文章《深度优先搜索》中介绍了深度优先搜索的相关知识，现在看一些实例，来加深理解运用。

以下问题中涉及到的树的结构如下

/**
* Definition for a binary tree node.
* public class TreeNode {
*     int val;
*     TreeNode left;
*     TreeNode right;
*     TreeNode(int x) { val = x; }
* }
*/

1.Validate Binary Search Tree

Given a binary tree, determine if it is a valid binary search tree (BST).

如果中序遍历的结果是一个递增序列，就说明它是一棵二叉查找树。因此，我们可以保存中序遍历的结果，然后对结果进行判断是否递增即可。

实际上，不必保存中序遍历的全部结果，只要能够保存上一个节点，并与当前的比较即可。

代码如下：

public class Solution {
    TreeNode pre;
    public boolean isValidBST(TreeNode root) {
        if(root == null) return true;
        if(!isValidBST(root.left)) return false;
        if(pre != null && pre.val >= root.val) return false;
        pre = root;
        if(!isValidBST(root.right)) return false;
        return true;
    }
}

2.Recover Binary Search Tree

Two elements of a binary search tree (BST) are swapped by mistake.

Recover the tree without changing its structure.

Note:
A solution using O(n) space is pretty straight forward. Could you devise a constant space solution?

通过分析发现，如果两个元素发生交换，那么中序遍历结果中会有一个或两个位置出现递减。当相邻两元素交换时，就只有一个递减位置。因此如果能用O(n)空间的话，直接保存中序遍历结果，并对结果进行遍历找到递减位置后交换即可。注意，两个元素分别是第一次出现递减的第一个元素，与第二次出现递减的第二个元素（如果不相邻两个元素发生交换的话）

如果只能用常量空间呢？这个思路可以参考上一道题，只要用两个变量分别来记录递减位置即可。

代码如下:

public class Solution {
    TreeNode pre = null;
    TreeNode first = null;
    TreeNode second = null;
    public void recoverTree(TreeNode root) {
        if(root == null) return ;
        helper(root);
        int tmp = first.val;
        first.val = second.val;
        second.val = tmp;
    }
    public void helper(TreeNode root){
        if(root == null) return;
        helper(root.left);
        if(pre == null){pre = root;}
        else{
            if(pre.val>root.val){
                if(first == null) first = pre;
                second = root;
            }
            pre = root;
        }
        helper(root.right);
    }
}