二叉搜索树的第k个节点

Posted 2020-12-03 樱圃

题目：给定一棵二叉搜索树，请找出其中的第k大的节点。

思路：如果按照中序遍历的顺序遍历一棵二叉搜索树，遍历序列的数值是递增排序的。因此，只需要用中序遍历算法遍历一棵二叉搜索树，就很容易找出它的第k大节点。

 ArrayList<TreeNode> arr = new ArrayList<TreeNode>();
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot==null|| k==0){
            return null;
        }
        inOrder(pRoot);
        TreeNode tr = pRoot;
        if(k<=arr.size()){
            tr = arr.get(k-1);
        }else{
            return null;
        }
        return tr;
    }

    public void inOrder(TreeNode root){
        if(root ==null){
            return;
        }
        if(root.left != null){
            inOrder(root.left);
        }
        arr.add(root);
        if(root.right != null){
            inOrder(root.right);
        }
    }