题解 | #二叉搜索树与双向链表#

二叉搜索树与双向链表

http://www.nowcoder.com/practice/947f6eb80d944a84850b0538bf0ec3a5

非递归(中序遍历),二叉搜索树的中序遍历就是得到它的从小到大排序数列,只需要在每个节点进行树结构和双向链表的转换就可以了,注意:要保存好头节点!

public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
import java.util.Stack;
public class Solution {
    private TreeNode cur = null;    //遍历的当前节点
    private TreeNode head = null;    //生成链表的头节点
    public TreeNode Convert(TreeNode pRootOfTree) {
        if(pRootOfTree==null)    return null;
        //非递归(中序遍历)
        Stack<TreeNode> stack = new Stack<>();
        TreeNode p = pRootOfTree;
        TreeNode pre = null;
        //寻找链表头节点
        while(p != null){
                stack.push(p);
                p = p.left;
            }
        p = stack.pop();
        //赋值链表头节点
        pRootOfTree = p;
        pre = p;
        p = p.right;
        //中序遍历树节点
        while(p != null || !stack.isEmpty()){
            while(p != null){
                stack.push(p);
                p = p.left;
            }
            p = stack.pop();
            //节点转化
            p.left = pre;
            pre.right = p;
            pre = p;
            //更新当前节点
            p = p.right;
        }
        return pRootOfTree;
    }
}

这样再原树结构上操作,空间复杂度为O(1), 遍历树的所有节点,时间复杂度为O(n),相比较递归的方法更优,不需要额外的空间保存递归调用栈。

全部评论
stack的空间不算空间复杂度吗
点赞 回复 分享
发布于 2022-06-16 14:53
要求不能生成新的任何节点
点赞 回复 分享
发布于 2022-07-26 22:23

相关推荐

10-30 22:18
已编辑
毛坦厂中学 C++
点赞 评论 收藏
分享
评论
4
收藏
分享
牛客网
牛客企业服务