前言
简单题,好难哈哈哈哈,加油干
分享一张图片思路,图片来源
题目
源码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public TreeNode trimBST(TreeNode root, int low, int high) {
if(root==null){
return null;
}
// ==========方法1,递归==========
// 由于二叉搜索树的特性,当大于high时,他的下一波肯定在左孩子中。递归获得符合的节点
// if(root.val>high){
// return trimBST(root.left,low,high);
// }
// // 小于low时一定在右孩子
// if(root.val<low){
// return trimBST(root.right,low,high);
// }
// // 上述得到符合的节点,就将符合的节点接入
// root.left=trimBST(root.left,low,high);
// root.right=trimBST(root.right,low,high);
// return root;
// ==========方法1,递归==========
// ============方法2,迭代========
while(root.val>high || root.val<low){
if(root.val>high){
root=root.left;
}else{
root=root.right;
}
}
TreeNode cur=root;
while(cur!=null){
while(cur.left!=null && cur.left.val<low){
cur.left=cur.left.right;
}
cur=cur.left;
}
cur=root;
while(cur!=null){
while(cur.right!=null&& cur.right.val>high){
cur.right=cur.right.left;
}
cur=cur.right;
}
return root;
// ============方法2,迭代========
}
}