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

package main
import . "nc_tools"
/*
 * type TreeNode struct {
 *   Val int
 *   Left *TreeNode
 *   Right *TreeNode
 * }
 */

/**
 * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
 *
 *
 * @param proot TreeNode类
 * @param k int整型
 * @return int整型
 */
func KthNode( proot *TreeNode ,  k int ) int {
    // write code here
    lenTree := 0
    getSize(proot, &lenTree)
    if proot == nil || k > lenTree || k <= 0{
        return -1
    }
    res := 0
    inOrder(proot, &k, &res)
    return res
}

func inOrder(proot *TreeNode, k *int, res *int) {    // 分析知本题应使用中序遍历解题
    if proot == nil {	// 1.如果当前节点为空,则直接返回
        return
    }
    if proot.Left != nil {	// 2.否则,若当前节点的左节点存在,中序遍历左节点(即一直遍历到整棵树的最左边)
        inOrder(proot.Left, k, res)
    }
    (*k)--        // 3.到此对k减1,表明访问到了第一小的元素
    if *k == 0 {    // 4.判断k是否为0,若是,说明当前节点即为第k小的元素,将值赋给res,并直接返回
        *res = proot.Val
        return
    }
    if proot.Right != nil {    // 5.若k不为0,说明当前节点不是第k小的元素,中序遍历右节点
        inOrder(proot.Right, k, res)
    }
}

func getSize(proot *TreeNode, lenTree *int) {	// 遍历所有元素来统计该树的节点个数,用于与k比较
    if proot != nil {
        *lenTree++
        getSize(proot.Left, lenTree)
        getSize(proot.Right, lenTree)
    }
}


全部评论

相关推荐

06-27 12:54
已编辑
门头沟学院 Java
累了,讲讲我的大学经历吧,目前在家待业。我是一个二本院校软件工程专业。最开始选专业是觉得计算机感兴趣,所以选择了他。本人学习计算机是从大二暑假结束开始的,也就是大三开始。当时每天学习,我个人认为Java以及是我生活的一部分了,就这样持续学习了一年半,来到了大四上学期末,大概是在12月中旬,我终于找的到了一家上海中厂的实习,但我发现实习生的工作很枯燥,公司分配的活也不多,大多时间也是自己在自学。就这样我秋招末才找到实习。时间来到了3月中旬,公司说我可以转正,但是转正工资只有7000,不过很稳定,不加班,双休,因为要回学校参加答辩了,同时当时也是心高气傲,认为可以找到更好的,所以放弃了转正机会,回学校准备论文。准备论文期间就也没有投递简历。然后时间来到了5月中旬,这时春招基本也结束了,然后我开始投递简历,期间只是约到了几家下场面试。工资也只有6-7k,到现在我不知道该怎么办了。已经没有当初学习的心劲了,好累呀,但是又不知道该干什么去。在家就是打游戏,boss简历投一投。每天日重一次。26秋招都说是针对26届的人,25怎么办。我好绝望。要不要参加考公、考研、央国企这些的。有没有大佬可以帮帮我。为什么感觉别人找工作都是顺其自然的事情,我感觉自己每一步都在艰难追赶。八股文背了又忘背了又忘,我每次都花很长时间去理解他,可是现在感觉八股、项目都忘完了。真的已经没有力气再去学习了。图片是我的简历,有没有大哥可以指正一下,或者说我应该走哪条路,有点不想在找工作了。
码客明:太累了就休息一下兄弟,人生不会完蛋的
如果实习可以转正,你会不...
点赞 评论 收藏
分享
nus2201602...:兄弟,你这个简历撕了丢了吧,就是一坨,去找几个项目,理解项目流程,看几遍就是你的了,看看八股就去干了,多看看牛客里别人发出来的简历,对着写,你这写的啥啊,纯一坨
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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