二叉搜索树第K小元素

1.思路:迭代法

计算出二叉树左节点的个数,如果左节点的个数等于k-1;则根节点就是我们要找的值,如果左节点的个数大于k-1,表明我们要查找的第k个最小元素在左子树中,如果左节点的个数小于k-1,表明要查找的第k个最小元素在右子节点中,找到右子树中第k-count(left)-1个元素就是我们要找的元素

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int kthSmallest(TreeNode root, int k) {
        /*
        迭代
        */
        int leftcon=count(root.left);
        if(leftcon==k-1) return root.val;
        if(leftcon>k-1) return  kthSmallest(root.left, k);
        return  kthSmallest( root.right, k-leftcon-1);
    }
    public int count(TreeNode root){
        if(root==null) return 0;
        return 1+count(root.left)+count(root.right);
    }
}

2.中序遍历;中序遍历二叉搜索树,排序是一个从小到大的数组

class Solution {
    private int cnt=0;
    private int val;
    public int kthSmallest(TreeNode root, int k) {
       //中序遍历
        inOrder(root,k);
        return val;
}
    public void inOrder(TreeNode node,int k){
        if(node==null) return;
        inOrder(node.left,k);
        cnt++;
        if(cnt==k){
            val=node.val;
            return;
        }
        inOrder(node.right,k);
    }
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
首先,我们可以按照中序遍历的方式遍历二叉搜索树,得到一个有序的节点序列。接下来,通过二分查找的方法寻找第一个大于等于k的元素的位置,然后又能得到需要寻找的比k小的元素的范围。最后,我们可以返回这个范围的所有元素即可。以下是使用Python实现的代码示例: ```python class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right def inorderTraversal(root): res = [] stack = [] curr = root while curr or stack: while curr: stack.append(curr) curr = curr.left curr = stack.pop() res.append(curr.val) curr = curr.right return res def findElementsLessThanK(root, k): if not root: return None nodes = inorderTraversal(root) l, r = 0, len(nodes) - 1 while l <= r: mid = (l + r) // 2 if nodes[mid] >= k: r = mid - 1 else: l = mid + 1 return nodes[:l] # 示例 root = TreeNode(5) root.left = TreeNode(3) root.right = TreeNode(7) root.left.left = TreeNode(2) root.left.right = TreeNode(4) root.right.left = TreeNode(6) root.right.right = TreeNode(8) k = 6 print(findElementsLessThanK(root, k)) # [2, 3, 4, 5] ``` 该代码中,`inorderTraversal`函数实现了中序遍历,返回节点升序排列的列表。`findElementsLessThanK`函数将中序遍历的结果进行二分查找,获得第一个大于等于k的位置`l`,然后返回节点列表中前`l`个元素,即为比k小的元素集合。最后,我们提供了一个示例,其中我们构建了一个二叉搜索树,然后通过该函数找到了所有比k小的元素

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值