【二叉树】二叉树展开为链表-力扣 114 题

news2024/11/27 11:33:04

💝💝💝欢迎来到我的博客，很高兴能够在这里和您见面！希望您在这里可以感受到一份轻松愉快的氛围，不仅可以获得有趣的内容和知识，也可以畅所欲言、分享您的想法和见解。

推荐:kuan 的首页,持续学习,不断总结,共同进步,活到老学到老
导航
檀越剑指大厂系列:全面总结 java 核心技术点,如集合,jvm,并发编程 redis,kafka,Spring,微服务,Netty 等
常用开发工具系列:罗列常用的开发工具,如 IDEA,Mac,Alfred,electerm,Git,typora,apifox 等
数据库系列:详细总结了常用数据库 mysql 技术点,以及工作中遇到的 mysql 问题等
懒人运维系列:总结好用的命令,解放双手不香吗?能用一个命令完成绝不用两个操作
数据结构与算法系列:总结数据结构和算法,不同类型针对性训练,提升编程思维,剑指大厂

非常期待和您一起在这个小小的网络世界里共同探索、学习和成长。💝💝💝 ✨✨ 欢迎订阅本专栏 ✨✨

博客目录

给你二叉树的根结点 root ，请你将它展开为一个单链表：

展开后的单链表应该同样使用 TreeNode ，其中 right 子指针指向链表中下一个结点，而左子指针始终为 null 。
展开后的单链表应该与二叉树先序遍历顺序相同。

输入：root = [1,2,5,3,4,null,6]
输出：[1,null,2,null,3,null,4,null,5,null,6]

题解1:

public void flatten(TreeNode root) {
    //先序遍历
    List<TreeNode> list = new ArrayList<>();
    LinkedList<TreeNode> stack = new LinkedList<>();
    TreeNode curr = root;
    TreeNode pop = null;
    TreeNode  p = null;
    while (curr != null || !stack.isEmpty()) {
        if (curr != null) {
            list.add(curr);
            stack.push(curr);
            curr = curr.right;
        } else {
            final TreeNode peek = stack.peek();
            if (peek.left == null || peek.left == pop) {
                pop = stack.pop();
            } else {
                curr = peek.left;
            }
        }
    }
}

题解2:

public void flatten(TreeNode root) {
    LinkedList<TreeNode> stack = new LinkedList<>();
    TreeNode curr = root;
    TreeNode pre = null;
    while (curr != null || !stack.isEmpty()) {
        while (curr != null) {
            stack.push(curr);
            curr = curr.right;
        }
        curr = stack.peek();
        if (curr.left == null || curr.left == pre) {
            //弹出的节点
            curr = stack.pop();
            curr.left = null;
            //第一次右节点为null
            curr.right = pre;
            //把当前节点赋值为弹出节点
            pre = curr;
            curr = null;//为了弹出下一个元素
        } else {
            curr = curr.left;
        }
    }
}