力扣刷题day17|654最大二叉树、617合并二叉树、700二叉搜索树中的搜索、98验证二叉搜索树

654. 最大二叉树

力扣题目地址

给定一个不重复的整数数组 nums最大二叉树 可以用下面的算法从 nums 递归地构建:

  1. 创建一个根节点,其值为 nums 中的最大值。
  2. 递归地在最大值 左边子数组前缀上 构建左子树。
  3. 递归地在最大值 右边子数组后缀上 构建右子树。

返回 nums 构建的 *最大二叉树*

示例1:

输入:nums = [3,2,1,6,0,5]
输出:[6,3,5,null,2,0,null,null,1]
解释:递归调用如下所示:
- [3,2,1,6,0,5] 中的最大值是 6 ,左边部分是 [3,2,1] ,右边部分是 [0,5] 。
    - [3,2,1] 中的最大值是 3 ,左边部分是 [] ,右边部分是 [2,1] 。
        - 空数组,无子节点。
        - [2,1] 中的最大值是 2 ,左边部分是 [] ,右边部分是 [1] 。
            - 空数组,无子节点。
            - 只有一个元素,所以子节点是一个值为 1 的节点。
    - [0,5] 中的最大值是 5 ,左边部分是 [0] ,右边部分是 [] 。
        - 只有一个元素,所以子节点是一个值为 0 的节点。
        - 空数组,无子节点。

示例 2:

输入:nums = [3,2,1]
输出:[3,null,2,null,1]

递归思路

递归三要素

  1. 确定递归函数的参数和返回值:参数就是传入的是存放元素的数组,和该数组的左右区间,返回该数组构造的二叉树的头结点,返回类型是指向节点的指针。
public TreeNode traversal(int[] nums, int left, int right)
  1. 确定终止条件:如果最后数组大小为1,说明遍历到了叶子节点了,因为区间左闭右开。左边界等于右边界就返回null
// 最后分组没有元素了就返回
if (right - left == 0) {
    return null;
}
  1. 确定单层递归的逻辑:先找到最大值确定根节点,然后分割左右数组
  • 找到数组中最大的值和对应的下标, 最大的值构造根节点,下标用来下一步分割数组。
// 找到最大值分割
int maxIndex = left;
for (int i = left + 1; i < right; i++) {
    if (nums[i] > nums[maxIndex]) {
        maxIndex = i;
    }
}
// 用最大值建立根节点
TreeNode root = new TreeNode();
root.val = nums[maxIndex];
  • 构造左右子树
// 分割后如果左右组都还有数值
// 左闭右开:[left, maxValueIndex)
root.left = traversal(nums, left, maxIndex);
// 左闭右开:[maxValueIndex + 1, right)
root.right = traversal(nums, maxIndex + 1, right);

完整代码:

public TreeNode constructMaximumBinaryTree(int[] nums) {

    return traversal(nums, 0, nums.length);
}

public TreeNode traversal(int[] nums, int left, int right) {
    // 最后分组没有元素了就返回
    if (right - left == 0) {
        return null;
    }
    // 找到最大值分割
    int maxIndex = left;
    for (int i = left + 1; i < right; i++) {
        if (nums[i] > nums[maxIndex]) {
            maxIndex = i;
        }
    }
    // 用最大值建立根节点
    TreeNode root = new TreeNode();
    root.val = nums[maxIndex];

    // 分割后如果左右组都还有数值
    // 左闭右开:[left, maxValueIndex)
    root.left = traversal(nums, left, maxIndex);
    // 左闭右开:[maxValueIndex + 1, right)
    root.right = traversal(nums, maxIndex + 1, right);

    return root;

}

617. 合并二叉树

力扣题目链接

给你两棵二叉树: root1root2

想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。

返回合并后的二叉树。

注意: 合并过程必须从两个树的根节点开始。

示例 1:

输入:root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出:[3,4,5,5,4,null,7]

示例 2:

输入:root1 = [1], root2 = [1,2]
输出:[2,2]

前序遍历递归思路

同时传入两个节点进行操作

递归三要素

  1. 确定递归函数的参数和返回值:要合入两个二叉树,那么参数至少是要传入两个二叉树的根节点,返回值就是合并之后二叉树的根节点
public TreeNode mergeTrees(TreeNode root1, TreeNode root2)
  1. 确定终止条件:用两个树节点t1、t2来遍历两个树

    如果root1== NULL 了,两个树合并就应该是 root2 (如果root2也为NULL也无所谓,合并之后就是NULL)。

    反过来如果root2 == NULL,那么两个数合并就是root1(如果root1也为NULL也无所谓,合并之后就是NULL)。

如图所示,此时将传进mergeTrees(root1.left, root.left),进行终止条件if (t1 == NULL) return t2;

当root1.left为空时,root2.left也是空,因此返回null,合并后该节点的左子树为空

然后mergeTrees(root1.right, root.right),进行终止条件if (t1 == NULL) return t2;

当root1.right为空时,root2.right不为空,因此返回当前的root2,合并后该节点就不用管root1了,直接将root2.right接上就行

image-20221009150417403

if (root1 == null) {
    return root2;
}
if (root2 == null) {
    return root1;
}
  1. 确定单层递归的逻辑:把root1当作合并后的新树,root1就是合并之后树的根节点(就是修改了原来树的结构)。

那么单层递归中,就要把两棵树的元素加到一起。

接下来root1 的左子树是:合并 root1左子树 root2左子树之后的左子树。

root1 的右子树:是 合并 root1右子树 root2右子树之后的右子树。

最终root1就是合并之后的根节点。

root1.val += root2.val;
root1.left = mergeTrees(root1.left, root2.left);
root1.right = mergeTrees(root1.right, root2.right);
return root1;

完整代码:

public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
    if (root1 == null) {
        return root2;
    }
    if (root2 == null) {
        return root1;
    }
    root1.val += root2.val;
    root1.left = mergeTrees(root1.left, root2.left);
    root1.right = mergeTrees(root1.right, root2.right);
    return root1;
}

层序遍历迭代

不用考虑if (node1.left != null && node2.left == null),因为最后返回的是root1,node2为空直接忽略

// 迭代---------------------------------------------------------------
public TreeNode mergeTrees1(TreeNode root1, TreeNode root2) {
    if (root1 == null) {
        return root2;
    }
    if (root2 == null) {
        return root1;
    }
    Deque<TreeNode> que = new ArrayDeque<>();
    que.offer(root1);
    que.offer(root2);
    while (!que.isEmpty()) {
        TreeNode node1 = que.poll();
        TreeNode node2 = que.poll();
        node1.val += node2.val;

        // 如果两棵树左节点都不为空,加入队列
        if (node1.left != null && node2.left != null) {
            que.offer(node1.left);
            que.offer(node2.left);
        }
        // 如果两棵树右节点都不为空,加入队列
        if (node1.right != null && node2.right != null) {
            que.offer(node1.right);
            que.offer(node2.right);
        }
        // 若node1的左节点为空,直接赋值
        if (node1.left == null && node2.left != null) {
            node1.left = node2.left;
        }
        // 若node1的右节点为空,直接赋值
        if (node1.right == null && node2.right != null) {
            node1.right = node2.right;
        }
    }

    return root1;
}
// 迭代---------------------------------------------------------------

700. 二叉搜索树中的搜索

力扣题目地址

给定二叉搜索树(BST)的根节点 root 和一个整数值 val

你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 null

示例 1:

输入:root = [4,2,7,1,3], val = 2
输出:[2,1,3]

示例 2:

输入:root = [4,2,7,1,3], val = 5
输出:[]

递归思路

递归三要素

  1. 确定递归函数的参数和返回值:递归函数的参数传入的就是根节点和要搜索的数值,返回的就是以这个搜索数值所在的节点。
public TreeNode searchBST(TreeNode root, int val)
  1. 确定终止条件:如果遍历到目标值或者root遍历完子树到null都没找到,就返回root节点
if (root == null || root.val == val) {
    return root;
}
  1. 确定单层递归的逻辑:因为二叉搜索树的节点是有序的,所以可以有方向的去搜索。

​ 如果root.val > val,搜索左子树,如果root.val < val,就搜索右子树,最后如果都没有搜索到,就返回NULL。

if (root.val > val) return searchBST(root.left, val);
if (root.val < val) return searchBST(root.right, val);

完整代码:

// 递归---------------------------------------------------------------
public TreeNode searchBST(TreeNode root, int val) {
    if (root == null || root.val == val) {
        return root;
    }
    if (root.val > val) return searchBST(root.left, val);
    if (root.val < val) return searchBST(root.right, val);
    return null;
}
// 递归---------------------------------------------------------------

迭代思路

对于一般二叉树,递归过程中还有回溯的过程,例如走一个左方向的分支走到头了,那么要调头,在走右分支。

对于二叉搜索树,不需要回溯的过程,因为节点的有序性就帮我们确定了搜索的方向。

完整代码:

// 迭代---------------------------------------------------------------
public TreeNode searchBST1(TreeNode root, int val) {
    while (root != null) {
        if (root.val > val) {
            root = root.left;
        }else if (root.val < val) {
            root = root.right;
        }else {
            return root;
        }
    }

    return null;
}

98. 验证二叉搜索树

力扣题目链接

给你一个二叉树的根节点 root ,判断其是否是一个有效的二叉搜索树。

有效 二叉搜索树定义如下:

  • 节点的左子树只包含 小于 当前节点的数。

  • 节点的右子树只包含 大于 当前节点的数。

  • 所有左子树和右子树自身必须也是二叉搜索树。

示例 1:

输入:root = [2,1,3]
输出:true

示例 2:

输入:root = [5,1,4,null,null,3,6]
输出:false
解释:根节点的值是 5 ,但是右子节点的值是 4 。

递归思路

要知道中序遍历下,输出的二叉搜索树节点的数值是有序序列。

有了这个特性,验证二叉搜索树,就相当于变成了判断一个序列是不是递增的了。

我们要比较的是左子树所有节点小于中间节点,右子树所有节点大于中间节点

递归三要素:

  1. 确定递归函数的参数和返回值:传入的参数就是根节点,当寻找某一条边(或者一个节点)的时候,递归函数会有bool类型的返回值
public boolean isValidBST(TreeNode root)
  1. 确定终止条件:如果是空节点也是二叉搜索树
// 最极端情况
if (root == null) {
    return true;
}
  1. 确定单层递归的逻辑:中序遍历,一直更新maxVal,一旦发现maxVal >= root.val,就返回false,注意元素相同时候也要返回false
// 中序遍历,左中右
boolean left = isValidBST(root.left); // 左
if (maxVal < root.val) { // 中
    maxVal = root.val;
}else {
    return false;
}
boolean right = isValidBST(root.right); // 右
难点
  • 不能单纯的比较左节点小于中间节点,右节点大于中间节点就完事了
if (root.val > root.left->val && root.val < root.right.val) {
    return true;
} else {
    return false;
}

我们要比较的是 左子树所有节点小于中间节点,右子树所有节点大于中间节点。所以以上代码的判断逻辑是错误的。

如图6出现在根节点10的右子树是不符合题意的

image-20221010205629067

  • 定义初始全局的最大值时要用long maxVal = Long.MIN_VALUE,因为后台给定的节点val包含了Integer.MIN_VALUE

完整代码:

// 递归---------------------------------------------------------------
long maxVal = Long.MIN_VALUE;
public boolean isValidBST(TreeNode root) {
    // 最极端情况
    if (root == null) {
        return true;
    }

    // 中序遍历,左中右
    boolean left = isValidBST(root.left); // 左
    if (maxVal < root.val) { // 中
        maxVal = root.val;
    }else {
        return false;
    }
    boolean right = isValidBST(root.right); // 右

    return left && right;
}

双指针递归

不初始化最小值,用一个pre指针指向左边的数值来和root比较

// 双指针递归---------------------------------------------------------------
TreeNode pre;
public boolean isValidBST1(TreeNode root) {
    // 最极端情况
    if (root == null) {
        return true;
    }
    boolean left = isValidBST1(root.left); // 左
    if (pre != null && pre.val >= root.val) {
        return false;
    }
    // 记录前一个节点
    pre = root;
    boolean right = isValidBST1(root.right);
    return left && right;

}

用栈迭代

// 迭代---------------------------------------------------------------
public boolean isValidBST2(TreeNode root) {
    if (root == null) {
        return true;
    }
    TreeNode pre = null;
    Stack<TreeNode> stack = new Stack<>();
    while (root != null || !stack.isEmpty()) {
        while (root != null) {
            stack.push(root);
            root = root.left; // 左
        }

        // 中
        TreeNode node = stack.pop();
        if (pre != null && node.val <= pre.val) {
            return false;
        }
        pre = node;

        root = node.right;
    }

    return true;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值