二叉树的第K个节点(中序遍历递归&非递归)

二叉搜索树的第k个结点

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

栈非递归算法:

class Solution {
public:
    TreeNode* KthNode(TreeNode* pRoot, int k) {
        if(!pRoot||k <= 0){ // pRoot为空返回 k<=0 也返回
            return nullptr;
        }
        stack<TreeNode *> s;
        TreeNode *cur = pRoot;
        while(!s.empty()||cur){
            if(cur){
                s.push(cur);
                cur = cur->left;
            }else{
                cur = s.top();
                s.pop();
                if(--k==0){
                    return cur;
                }
                cur = cur->right;
            }
        }
            return nullptr;
   }
};

递归算法:

class Solution {
public:
    int f = 1;
    TreeNode *ans = NULL;
    TreeNode* KthNode(TreeNode* pRoot, int k) {
        if(!pRoot||k<=0)return NULL;
        inOrder(pRoot, k);
        return ans;
    }
    void inOrder(TreeNode *pRoot,int k){
        if(!pRoot)return;
        inOrder(pRoot->left, k);
        if(f++==k)ans= pRoot; // 按照中序遍历顺序进行查找找到赋值
        inOrder(pRoot->right, k);
    }
};
全部评论

相关推荐

06-26 15:33
青岛工学院 Java
积极的秋田犬要冲国企:他现在邀请我明天面试
点赞 评论 收藏
分享
06-15 18:44
黄淮学院 Java
Lynn012:如果是居民楼还是算了吧,看着有点野呢
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

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