A股上市公司传智教育(股票代码 003032)旗下技术交流社区北京昌平校区

 找回密码
 加入黑马

QQ登录

只需一步,快速开始

题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8)    中,按结点数值大小顺序第三小结点的值为4。

分析
常见方法是用中序遍历算法来做,中序遍历时,输出顺序正好是由小到大。

代码
import java.util.Stack;

public class KthNode {

       
        public static int index = 0;
        // 递归版本
        public static TreeNode kthNode1(TreeNode root, int k) {
                if (root != null) {
                        TreeNode node = kthNode1(root.left, k);
                        if (node != null) {
                                return node;
                        }
                        index++;
                        if (index == k) {
                                return root;
                        }
                        node = kthNode1(root.right, k);
                        if (node != null) {
                                return node;
                        }
                }
                return null;
        }

        // 非递归版本
        public static TreeNode kthNode2(TreeNode root, int k) {
                if (root == null || k == 0)
                        return null;
                int count = 0;
                Stack<TreeNode> stack = new Stack<>();
                while (root != null || !stack.isEmpty()) {
                        while (root != null) {
                                stack.push(root);
                                root = root.left;
                        }
                        root = stack.pop();
                        count++;
                        if (count == k)
                                return root;
                        root = root.right;
                }
                return null;
        }

}

---------------------
作者:另一个我竟然存在
来源:CSDN
原文:https://blog.csdn.net/qq_24034545/article/details/83660147
版权声明:本文为博主原创文章,转载请附上博文链接!

1 个回复

正序浏览
奈斯
回复 使用道具 举报
您需要登录后才可以回帖 登录 | 加入黑马