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

杀马特。学长 韩版系。学妹 提交于 2020-01-27 22:32:10

题目描述

给定一棵二叉搜索树,请找出其中的第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);
    }
}
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!