修剪二叉搜索树

修剪二叉搜索树

问题:

在这里插入图片描述
在这里插入图片描述

思路:

分治递归:

递归函数的功能是使以当前节点为根的树修剪完毕且返回处理完成后的树根节点

当根节点的值小于下界时,根节点以及其左子树应全部丢弃,此时返回对根节点的右孩子递归调用函数的返回值。当根节点的值大于上界时,根节点以及其右子树应全部丢弃,此时返回对根节点的左孩子递归调用函数的返回值。

若根节点的值处于下界与上界之间,分别对根节点的左右子树进行递归函数调用,其返回值作为根节点的左右孩子,最后返回根节点。

代码:

/**
 * 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 root;
        if( root->val<L )
            return trimBST( root->right,L,R );
        if( root->val>R )
            return trimBST( root->left,L,R );
        root->left=trimBST( root->left,L,R );
        root->right=trimBST( root->right,L,R );
        return root;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_44709990/article/details/105938742