码迷,mamicode.com
首页 > 其他好文 > 详细

Kth Smallest Element in a BST

时间:2015-07-14 15:01:24      阅读:82      评论:0      收藏:0      [点我收藏+]

标签:

class Solution {
public:
    int kthSmallest(TreeNode* root, int k) {
        int num = 0;
        TreeNode*pNode=NULL;
        visit(root, k, num, pNode);

        return pNode->val;
    }
    
private:
    void visit(TreeNode *root, int k, int &num, TreeNode*&pNode)
    {
        if (root == NULL)
        {
            return;
        }

        visit(root->left, k, num, pNode);
        
            
            if (++num == k)
            {
                pNode=root;
                return;
            }
        
        visit(root->right, k, num, pNode);
    }
};

此类方法遍历的时候,不要用返回值表示,如果有返回值,那么我们遍历到NULL时候,我们需要回退。此时如果带返回值,那么导致结果错误。

(只能使用不带返回值的void,这样我们才能回到上一层,且能继续执行下面的代码,否则我们将直接终止程序)

Kth Smallest Element in a BST

标签:

原文地址:http://www.cnblogs.com/kkshaq/p/4645177.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!