剑指offer 63题 【树】二叉搜索树的第k个结点

题目描述

给定一颗二叉搜索树，请找出其中的第k大的结点。例如， 5 3 7 2 4 6 8 中，按结点数值大小顺序第三个结点的值为4。

牛客传送门：点击打开链接

思路：

结合二叉搜索树的特点(左结点及其子树比当前节点小，右结点及其子树比当前节点大)。中序遍历即可知道第k大的结点。

测试用例：二叉搜索树：[5,3,7,2,4,6,8]     k数值为：2、5、7、9、-20

下面为代码：

public class Title63 {
    int curNumber;
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot == null)
            return null;
        // 注意初始化
        curNumber = 0;
        return getKthNode(pRoot, k);
    }
    
    TreeNode getKthNode(TreeNode curNode,int k){
        TreeNode ret = null;
        // 遍历左子树
        if(curNode.left != null){
            ret = getKthNode(curNode.left,k);
        }
        // 当前数字
        curNumber ++ ;
        if(curNumber == k){
            return curNode;
        }
        if(ret != null)
            return ret;
        // 遍历右子树
        if(curNode.right != null){
            ret = getKthNode(curNode.right,k);
        }
        return ret;
    }
    
    /**
     * 测试
     * 2016年7月22日
     */
    public static void main(String[] args) {
        TreeNode a = new TreeNode(5);
        TreeNode b = new TreeNode(3);
        TreeNode c = new TreeNode(7);
        TreeNode d = new TreeNode(2);
        TreeNode e = new TreeNode(4);
        TreeNode f = new TreeNode(6);
        TreeNode g = new TreeNode(8);
        a.left = b;a.right = c;
        b.left = d;b.right = e;
        c.left = f;c.right = g;

        Title63  clazz = new Title63();
        TreeNode node = clazz.KthNode(a, 2);
        System.out.println(node == null ? "null" : node.val);
        node = clazz.KthNode(a, 5);
        System.out.println(node == null ? "null" : node.val);
        node = clazz.KthNode(a, 7);
        System.out.println(node == null ? "null" : node.val);
        node = clazz.KthNode(a, 9);
        System.out.println(node == null ? "null" : node.val);
        node = clazz.KthNode(a, -20);
        System.out.println(node == null ? "null" : node.val);
        // 3   6   8   null   null
    }
}