题目地址:
https://leetcode.com/problems/trim-a-binary-search-tree/
给定一棵二叉搜索树,再给定一个范围 ,要求将这棵树所有在这个范围之外的节点都删掉,返回新树树根。
思路是用分治法。如果是空树则直接返回,否则看树根的范围,如果树根处于
,那说明树根以及整个左子树都要被砍掉,所以只需返回trim(root.right, L, R)
;如果树根处于
,那说明树根以及整个右子树都要被砍掉,所以只需返回trim(root.left, L, R)
;否则的话,说明树根不需要被砍掉,只需分而治之,把左右子树trim完的树根返回过去与原树根连上即可。代码如下:
public class Solution {
public TreeNode trimBST(TreeNode root, int L, int R) {
if (root == null) {
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;
}
}
}
class TreeNode {
int val;
TreeNode left, right;
TreeNode(int x) {
val = x;
}
}
时间复杂度 ,空间 。