leetcode: 二叉搜索树中第ｋ小的元素

题目来源:力扣

题目要求：

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

审题：

看到这道题，我一开始想到，由于二叉树的左子树节点值均小于当前节点值，因此当前节点由小到大排序的次序即是其其左子树节点个数．因此，如果当前节点的排名大于次序ｋ，我们便在左子树继续寻找，如果等于排名，即找到；否则在右子树中查找．考虑一下该算法的时间复杂度，其中节点最多搜索lgN次，而每次计算节点排名的时间复杂度则等于其子树大小．具体的时间复杂度推导较为繁琐．基于该方法的算法实现会在回文介绍．

上面这方法显然不是最好的，应为我们对于二叉树的本质特征没有充分利用？什么本质特征呢？那就是二叉树节点中序遍历得到的节点序列，其节点值从小到大依次排序．基于该特征，我们想到的第一种方法便是对二叉树节点进行中序遍历，然后获取该序列的第ｋ个值即可．进一步考虑，如果我们只是进行一次调用，那么保存二叉树节点的中序遍历是不必要的，我们只需要在对节点进行中序遍历时，返回第次序为ｋ的节点的值即可．后文介绍该算法的具体实现．

java算法实现

方法一：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    //计算当前节点为根节点的子树中节点个数
    private int numNodes(TreeNode x){
        if(x == null)
            return 0;
        else
            return 1 + numNodes(x.left) + numNodes(x.right);
    }

    public int kthSmallest(TreeNode root, int k) {
        while(root != null){
            int lessNum = numNodes(root.left);
            if(lessNum+1 == k) //如果找到
                return root.val;
            else if(lessNum >= k){//节点值大于目标值
                root = root.left;
            }
            else{
                root = root.right;
                k -= (lessNum+1);
            }
        }
        return -1;
    }
}

方法二：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {

    int val = -1;
    int order;

    private void inorder(TreeNode root, int k){
        if(root == null || order > k)
            return ;
        inorder(root.left, k);
        order++; //遍历当前元素
        if(order == k){//是当前元素
            val = root.val;
        }
        inorder(root.right, k);
    }

    //二叉搜索树中序遍历结果即为所有元素从小到大排序结果
    public int kthSmallest(TreeNode root, int k) {
        inorder(root, k);
        return val;
    }
}