剑指offer06

128 阅读1分钟

题目描述

给定一颗二叉搜索树,请找出其中的第K小的节点。例如,(5,3,7,2,46,8)中,按节点数值大小顺序第三小节点值为4

解题思路分析

其实要解这道题首先需要了解什么是二叉搜索树,如果不知道二叉搜索树的相关性质的话,那解答这道题就很难了。话说回来,二叉搜索树其实就是一颗二叉树拥有这样的特点:所有父节点都比其左子节点大,比其右子节点小。知道这个性质之后,这道题马上就可以想出来了,只要中序遍历一遍这棵树便可以得到一个排好序的序列,然后找第k小的节点就很简单了。

代码实现

public static TreeNode KthNode(TreeNode pRoot, int k) {
        if (pRoot == null || k <= 0) {
            return null;
        }
        ArrayList<TreeNode> list = new ArrayList<TreeNode>();
        // 中序遍历得到该二叉搜索树的中序遍历序列
        inOrder(pRoot, list);
        if (list.size() < k) {
            return null;
        }
        return list.get(k - 1);
    }
    private static void inOrder(TreeNode pRoot, ArrayList<TreeNode> list) {
        if (pRoot == null) {
            return;
        }
        inOrder(pRoot.left, list);
        list.add(pRoot);
        inOrder(pRoot.right, list);
    }