c++中序遍历,点到为止

二叉搜索树的第k个结点

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

class Solution {
public:
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if (!pRoot || k==0) return nullptr;
        array.clear();
        MidTrav(array, pRoot, k);
        if (array.size() < k) return nullptr;
        return array.back();
    }
private:
    vector<TreeNode*> array;
    void MidTrav(vector<TreeNode*>& array, TreeNode* root, int end)
    {
        if (!root || array.size() == end) return;
        MidTrav(array, root->left, end);
        if (array.size() == end) return;
        array.push_back(root);
        MidTrav(array, root->right, end);
    }
};
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务