二叉搜索树的第 k 个结点

2年前 (2022) 程序员胖胖胖虎阿
236 0 0

给定一棵二叉搜索树,请找出其中的第 k 小的结点

解题思路

根据二叉搜索树的性质,按照中序遍历的顺序打印出来正好就是排序好的顺序,所以,按照中序遍历顺序找到第 k个结点就是结果

在遍历过程中,如果发现有符合条件的结点,则直接递归向上返回,因此返回的 node 必然不为空,正好符合判断条件。如果不做 node 是否为 null 的判断,则目标结点会被其父结点所覆盖

public class Solution {
    
    int count = 1;
    TreeNode KthNode(TreeNode pRoot, int k) {
        if(pRoot != null) {
            TreeNode node = KthNode(pRoot.left, k);
            if(node != null) {
                return node;
            }
            if(count == k) {
                return pRoot;
            }
            count++;
            node = KthNode(pRoot.right, k);
            if(node != null) {
                return node;
            }
        }
        return null;
    }
}

版权声明:程序员胖胖胖虎阿 发表于 2022年10月6日 上午7:40。
转载请注明:二叉搜索树的第 k 个结点 | 胖虎的工具箱-编程导航

相关文章

暂无评论

暂无评论...