给定一个二叉搜索树,同时给定最小边界L
和最大边界 R
。通过修剪二叉搜索树,使得所有节点的值在[L, R]
中 (R>=L) 。你可能需要改变树的根节点,所以结果应当返回修剪好的二叉搜索树的新的根节点。
示例 1:
输入: 1 / 0 2 L = 1 R = 2 输出: 1 2
示例 2:
输入: 3 / 0 4 2 / 1 L = 1 R = 3 输出: 3 / 2 / 1
二叉树的基本操作系列
1 /** 2 * Definition for a binary tree node. 3 * struct TreeNode { 4 * int val; 5 * TreeNode *left; 6 * TreeNode *right; 7 * TreeNode(int x) : val(x), left(NULL), right(NULL) {} 8 * }; 9 */ 10 class Solution { 11 public: 12 TreeNode* trimBST(TreeNode* root, int L, int R) { 13 if(root == NULL) 14 return root; 15 if(root->val < L) 16 return trimBST(root->right,L,R); 17 if(root->val > R) 18 return trimBST(root->left,L,R); 19 20 root->right = trimBST(root->right,L,R); 21 root->left = trimBST(root->left,L,R); 22 return root; 23 } 24 };