题解 | #二叉搜索树的第k个节点#

二叉搜索树的第k个节点

https://www.nowcoder.com/practice/57aa0bab91884a10b5136ca2c087f8ff

思路是中序遍历,在递归里面先递归左子树再递归右子树,然后每遍历一次记录数量count,要是到了count==k,保存这个节点。
import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param proot TreeNode类 
     * @param k int整型 
     * @return int整型
     */

    //二叉搜索树,中序遍历就是从小到大排列
    private TreeNode res = null ;
    int count = 0;

    public void midOrder(TreeNode root,int k){
        if(root == null || count > k){
            return;
        }
        //先遍历左边的子树
        midOrder(root.left,k);
        count ++;
        if(count == k){
            //只记录第k个节点
            res = root;
        }
        //再遍历右子树
        midOrder(root.right,k);
    }

    public int KthNode (TreeNode proot, int k) {
        // write code here
        midOrder(proot,k);
        
        if(res != null){
            return res.val;
        }else{
            return -1;
        }

    }
}

剑指offer刷题总结 文章被收录于专栏

本专栏是本人刷剑指offer的记录总结,也欢迎大家评论区留言讨论交流~

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务