首页 > 其他分享 >二叉搜索树中第K小的元素

二叉搜索树中第K小的元素

时间:2022-12-30 23:00:49浏览次数:66  
标签:traverse return val root sum 二叉 搜索 null 树中

二叉搜索树中第K小的元素

https://leetcode.cn/problems/kth-smallest-element-in-a-bst/


解题思路:

  1. 生序排序,然后找第K个元素
  2. BST的中序遍历就是升序排序的结果

let  kthSmallest  = (root, k) => {
  // 递归遍历
let rank = 0;
let res = 0;
 function traverse(root, k){
    if(root == null){
      return
     }
    // 中序遍历 
   traverse(root.left, k)
   rank ++;
  if(rank === k){
    // 找到该值了,存起来
     res = root.val
     return
  }
  traverse(root.right, k)
 }
  traverse(root, k)
  return root
}
 return res;

把二叉搜索树转换为累加树

https://leetcode.cn/problems/convert-bst-to-greater-tree/

解题思路

  1. BST累加树,老套路,利用中序遍历,根据右边节点大于左边节点的特性,倒序
  2. 用一个变量来存储当前节点的累加和
function convertBST(root){
  let sum = 0
  function traverse(root){
    if(root == null){
      return 
    }
    traverse(root.right)
   sum += root.val;
   root.val = sum; // 将累加的结果赋给当前节点
    traverse(root.left)
 }
 traverse(root)
  return root
}

方法2: 将方法一转变一下,通过传参数达到用一个变量存储sum的效果

var convertBST = function(root){
  if(root == null){
    return null
  }
  traverse(root, 0)
}

function traverse(root, sum){
  if(root == null){
    return sum;
   }

  let sum2 =  traverse(root.right, sum)
   root.val += sum2;
   sum2 = root.val
  sum2 = traverse(root.left, sum2)
  return sum2
}

二叉搜索树的属性

https://leetcode.cn/problems/validate-binary-search-tree/

解题思路

  1. 根据BST框架,左小右大,存在max.val > root.val > min.val
 function isValidBST(root){
    function dfs(root, min, max){
    // 判断树为空的情况
      if(!root){
        return true
      }
   // 关键性的判断条件:左 < 根 < 右
    if((min != null && root.val <= min.val ) 
      || (max != null && root.val >= max.val) 
    ){
        return false
    }
   //  遍历左子树和右子树
     return  dfs(root.left, min, root) && dfs(root.right, root, max)
    }
   return dfs(root, null, null)

}

标签:traverse,return,val,root,sum,二叉,搜索,null,树中
From: https://www.cnblogs.com/yiyunh/p/17015830.html

相关文章