劍指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
    }
}