代码随想录二叉树——修剪二叉树

题目

给定一个二叉搜索树,同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树,使得所有节点的值在[L, R]中 (R>=L)
。你可能需要改变树的根节点,所以结果应当返回修剪好的二叉搜索树的新的根节点。

代码随想录二叉树——修剪二叉树_第1张图片
代码随想录二叉树——修剪二叉树_第2张图片

思路

递归法

  1. 递归函数的参数和返回值:根节点和需要修剪的区间

  2. 终止条件:遇到空节点返回即可,因为修剪的操作不是在终止条件上进行的

  3. 单层递归的逻辑:如果root的元素小于low 的值, 则递归右子树,并返回右子树符合条件的头结点;反之递归左子树

java代码如下:

class Solution {
	public TreeNode trimBST(TreeNode root, int low, int high) {
		if(root == null) return null;
		if(root.val < low){
			return trimBST(root.right,low,high);//继续在右子树寻找符合区间的节点
		}
		if(root.val > high){
			return trimBST(root.left,low,high);//继续在左子树寻找符合区间的节点
		}
		//如果root在[low,high]范围内
		root.left = trimBST(root.left,low,high);// root->left接入符合条件的左孩子,从而实现删除操作
		root.right = trimBST(root.right,low,high);// root->right接入符合条件的右孩子,从而实现删除操作
		return root;
	}
}

你可能感兴趣的:(代码随想录,算法)