【LeetCode】669. Trim a Binary Search Tree

修剪二叉树算法

最新推荐文章于 2025-02-03 15:37:10 发布

原创最新推荐文章于 2025-02-03 15:37:10 发布 · 296 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#代码待补全

本文介绍了一种修剪二叉树的算法，通过递归方式去除不在指定范围内的节点。当根节点值小于L时，保留右子树；大于R时，保留左子树。特殊情况考虑后，继续对左右子树进行遍历。

题目：修剪二叉树

题目要求：修剪掉在[L,R]之外的节点。

思路：递归

从根节点出发，若根节点 < L,则保留右子树，抛弃左子树

若根节点 > R ,则保留左子树，抛弃右子树

特殊情况考虑之后，则需要左右子树遍历

 struct TreeNode {
    int val;
    TreeNode *left;
	TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 };
 
class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int L, int R) {
        if(!root)
			return null;
		if(root -> val < L)
			return trimBST(root -> right, L, R);
		else if(root -> val > R)
			return trimBST(root -> left, L, R);
		else{
			root -> left = trimBST(root -> left, L, R);
			root -> right = trimBST(root -> right, L, R);
			return root;
		}
    }
};

（我现在还不会怎么输入二叉树来判断代码是否正确，等了解后再补全代码）