欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【LeetCode】669. Trim a Binary Search Tree

程序员文章站 2022-05-28 19:38:28
...

题目:修剪二叉树

【LeetCode】669. Trim a Binary Search Tree

题目要求:修剪掉在[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;
		}
    }
};
(我现在还不会怎么输入二叉树来判断代码是否正确,等了解后再补全代码)
相关标签: 代码待补全