非递归实现:二叉搜索树的第k个结点 中序遍历

二叉搜索树的第k个结点

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

import java.util.Stack;
public class Solution {
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot == null || k <= 0){
            return null;
        }
        Stack<TreeNode> stack = new Stack<>(); //建立栈
        TreeNode cur = pRoot;
        //while 部分为中序遍历
        while(!stack.isEmpty() || cur != null){ 
            if(cur != null){
                stack.push(cur); //当前节点不为null,应该寻找左儿子
                cur = cur.left;
            }else{
                cur = stack.pop();//当前节点null则弹出栈内元素,相当于按顺序输出最小值。
                if(--k == 0){ //计数器功能
                    return cur;
                }
                cur = cur.right;
            }
        }
        return null;
    }
}
全部评论
很明显空间超了。我的解法,时间空间应该都是可以的,缺陷是使用了全局变量。 class Solution { public: TreeNode* res; int n = 0; TreeNode* KthNode(TreeNode* pRoot, int k) { SearchNode(pRoot, k); return res; } void SearchNode(TreeNode* pRoot, int k){ if(pRoot == NULL){ return; } KthNode(pRoot->left, k); n++; if(n == k){ res = pRoot; } KthNode(pRoot->right, k); } };
点赞 回复 分享
发布于 2021-10-08 23:04
求问大佬,如果使用栈等集合去遍历存节点的话,还属于是O(1)的空间复杂度吗?
点赞 回复 分享
发布于 2021-10-03 15:35

相关推荐

评论
45
1
分享

创作者周榜

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