二叉树的第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);
    }
};
全部评论

相关推荐

点赞 评论 收藏
分享
10-06 12:46
门头沟学院 Java
跨考小白:定时任务启动
点赞 评论 收藏
分享
1 收藏 评论
分享
牛客网
牛客企业服务