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

二叉搜索树的第k个结点

http://www.nowcoder.com/practice/ef068f602dde4d28aab2b210e859150a

三种做法
总体来说 采用非递归的效率比其他方式实现要好一点 占用内存稍微小一点
1、非递归,采用中序遍历,求出第k个节点

    vector<TreeNode *> tp;
    void inOrder(TreeNode *root) { //使用中序遍历得到有序数组
        if(!root)
            return;
        inOrder(root->left);
        tp.push_back(root);
        inOrder(root->right);
    }
    TreeNode* KthNode(TreeNode* pRoot, int k) {
        if(!pRoot)
            return NULL;
        inOrder(pRoot);
        for(int i = 0; i < tp.size(); i++) {
            if(i == k-1)
                return tp[i];
        }
        return NULL;
    }

2、递归做法,无返回值的

    TreeNode *tp = NULL; //保存最终的结果
    void midOrder(TreeNode *pRoot, int &k) { //也是要用引用
        if(!pRoot || tp) //tp不为空 直接返回  也是避免无谓
            return;

        midOrder(pRoot->left, k);
        k--;
        if(k == 0) {
            tp = pRoot;
            return;
        }
        midOrder(pRoot->right, k);
    }
    TreeNode* KthNode(TreeNode* pRoot, int k) {
       if(!pRoot)
           return NULL;
        midOrder(pRoot, k);
        return tp;
    }

3、递归做法,有返回值的

//也是递归使用中序遍历,当k为0时 找到节点  如果开始没有找到结果 递归返回的都是NULL
//一旦找到结果 就会把这个结果层层返回
TreeNode * getRes(TreeNode *pRoot, int &k) { //主要要用引用 保证递归的过程只能使用
//一个变量 因为每层递归的k值是不同的
        if(!pRoot)
            return NULL;
        TreeNode *left = getRes(pRoot->left, k);
        if(left){ //不为空 找到结果  及时返回  可以避免后面的递归
            return left;
        }
        k--;
        if(k == 0) {
            return pRoot;
        }
        TreeNode *right = getRes(pRoot->right, k);
        if(right){ //不为空 找到结果  及时返回  可以避免后面的递归
            return right;
        }
        return left ? left : right ? right : NULL; //这里主要是返回NULL
    }
    TreeNode* KthNode(TreeNode* pRoot, int k) {
       if(!pRoot)
           return NULL;
        int tk = k;
        return getRes(pRoot, k);
    }
全部评论

相关推荐

沉淀一会:**圣经 1.同学你面试评价不错,概率很大,请耐心等待;2.你的排名比较靠前,不要担心,耐心等待;3.问题不大,正在审批,不要着急签其他公司,等等我们!4.预计9月中下旬,安心过节;5.下周会有结果,请耐心等待下;6.可能国庆节前后,一有结果我马上通知你;7.预计10月中旬,再坚持一下;8.正在走流程,就这两天了;9.同学,结果我也不知道,你如果查到了也告诉我一声;10.同学你出线不明朗,建议签其他公司保底!11.同学你找了哪些公司,我也在找工作。
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务