《剑指offer》 二叉搜索树的第k个节点（java）

题目描述

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

解题思路

首先得搞明白二叉搜索树是个啥，下面是我找到的对于二叉搜索树的定义。

二叉查找树（Binary Search Tree），（又：二叉搜索树，二叉排序树）它或者是一棵空树，或者是具有下列性质的二叉树： 若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值； 若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值； 它的左、右子树也分别为二叉排序树。

如果还不怎么明白二叉搜索树是啥，可以看看下面这张图，左节点<根节点<右节点。
题目是让寻找第k小的节点，那么就可以先遍历二叉搜索树的左子树的左节点，再遍历左子树的根节点然后是左子树的右节点；左子树完了，再遍历右子树，一直到找到第k小的节点。左-中-右的遍历顺序就是我们学的中序遍历，因此这个题可以用中序遍历来做。每遍历一个非空的节点，k–，一直到我们要找的那个节点为止，k正好是0。
这个题需要注意的是，k是变化的，一开始我直接用k当作变量传进参数去了，结果总是有点意外，于是整了一个成员变量。如果是C++实现的话，使用int &k传到函数里面就行了。

AC代码

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public TreeNode res=null;
    public int a=0;
    TreeNode KthNode(TreeNode root, int k)
    {
        a=k;
        dfs(root);
        return res;
    }
    
    void dfs(TreeNode root){
            if(root==null) 
                return;
            //中序遍历
            dfs(root.left);
            a--;
            if(a==0) {
                res=root;
            }
            dfs(root.right);
    }
}