669. Trim a Binary Search Tree(python+cpp)

题目:

Given a binary search tree and the lowest and highest boundaries as L and R, trim the tree so that all its elements lies in [L, R] (R >= L).You might need to change the root of the tree, so the result should return the new root of the trimmed binary search tree.

Example 1:

Input: 
    1 
   / \   
  0   2
L = 1   R = 2
Output: 
    1
      \
       2 

Example 2:

Input: 
      3  
     / \  
    0   4   
     \
      2  
     /   
    1
 L = 1   R = 3
Output: 
     3
    /     
   2      
  /  
 1

解释:
重构二叉排序树,使得其值在所给范围之内。用递归来做。
BST的特点是父亲结点的左孩子都比它小,右孩子都比它大,所以如果父亲结点都比最小值小了,就不要左边所有的树了,仅仅返回右边的树的修剪的结果,同理,如果父亲结点都比最大值大了,就不要右边的树了,仅仅返回左边的修剪的结果。
左孩子等于左边修剪后的结果,右孩子等于右边修剪后的结果。
python代码:

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def trimBST(self, root, L, R):
        """
        :type root: TreeNode
        :type L: int
        :type R: int
        :rtype: TreeNode
        """
        if (root):
            if root.val<L:
                return self.trimBST(root.right,L,R)
            elif root.val>R:
                return self.trimBST(root.left,L,R)
            else:
                root.left=self.trimBST(root.left,L,R)
                root.right=self.trimBST(root.right,L,R)
            	return root 
        return None

c++代码:

/**
 * Definition for a binary tree node.
 * 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==NULL)
            return NULL;
        else 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;
        }
    }
};

总结:
看到二叉树的时候,概率是递归了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值