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

二叉搜索树的第k个节点

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

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param proot TreeNode类 
     * @param k int整型 
     * @return int整型
     */
    int Find(TreeNode *proot){
        if(proot==nullptr){
            return 0;
        }
        int t=1;
        t+=(Find(proot->left)+Find(proot->right));
        return t;
    }
    int KthNode(TreeNode* proot, int k) {
        if(proot==nullptr){
            return -1;
        }
        if(Find(proot)<k){
            return -1;
        }
        int f=Find(proot->left);
        if(f>=k){
            return KthNode(proot->left,k);
        }
        else if(f+1==k){
            return proot->val;
        }
        else{
            return KthNode(proot->right, k-f-1);
        }
    }
};

全部评论

相关推荐

10-15 09:13
已编辑
天津大学 soc前端设计
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务