Leetcode __669. 修剪二叉搜索树

问题描述

给定一个二叉搜索树，同时给定最小边界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

思路

• 二叉搜索树：中序遍历时，根节点为中间值，根节点的值大于左子树的值，根节点的值小于右子树的值；
• 给定一个范围，就好比一个窗口,窗口内是有效数据，序列根节点就是一个中间标识，看根节点在窗口的位置；若根节点在窗口的右侧，则R < root.val，则右子树都剪掉，直接递归左子树返回即可；若根节点在窗口的左侧，则root.val < L，则左子树不是有效数据，剪掉，递归右子树返回即可；若在中间，则要递归更新左右子树，最终返回根节点。

实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode trimBST(TreeNode root, int L, int R){
        if(root==null){
            return null;
        }
        if(root.val>R){
            return trimBST(root.left,L,R);//要返回，情况中的一种，不return，返回最先压入栈的元素
        }
        if(root.val<L){
            return trimBST(root.right,L,R);
        }
        root.left = trimBST(root.left,L,R);//更新左子树
        root.right = trimBST(root.right,L,R);
        return root;
    }
}

小结

刚开始的想法是，中序遍历，去除范围内的数据，重新递归构建二叉搜索树，但这种耗时运算比较多，不是好的方法；
现在的方法只递归修剪即可，与之前相比，少了重新遍历，创建数组的过程。