算法-二叉树展开单链表

这道题我们可以利用栈来做,利用栈先进后出的特性 每次先加入右节点再加入左节点,这样的话弹出的时候正好左节点在前面,右节点在后面满足题目要求。

然后至于是构造单链表,我们可以用一个prev节点 prev的left永远都是null 而prev的right永远都等于cur

因为每次curr都是栈内弹出来的 按照左节点后进先出,所以prev的右节点每次其实都是原来二叉树的左节点,这样的话就能保证满足原二叉树的先序遍历。

代码如下:

java 复制代码
public class flatten {
    class  Solution{
        public void flatten(TreeNode root) {
         // 利用栈来解决 每次从栈内弹出一个节点作为当前访问的节点,获得该节点的子节点,如果子节点不为空,则以此右左节点压入栈内
            if(root == null) return;
            Deque<TreeNode> stack = new LinkedList<TreeNode>();
            stack.push(root);
            TreeNode prev = null;
            while(!stack.isEmpty()) {
                TreeNode curr=stack.pop();
                if(prev != null) {
                    prev.left=null;
                    prev.right=curr;
                }
                TreeNode left=curr.left;
                TreeNode right=curr.right;
                if(right != null) {
                    stack.push(right);
                }
                if(left != null) {
                    stack.push(left);
                }
                prev=curr;
            }


        }
    }
}
相关推荐
Jerry1 小时前
LeetCode 459. 重复的子字符串
算法
海石4 小时前
1500分的题目,确实有实力,不过还是我略胜一筹
算法·leetcode
海石4 小时前
【记忆化搜索】条条大路通AC,走好适合你的那一条,走到后再考虑走得快
算法·leetcode
Jerry6 小时前
LeetCode 151. 反转字符串中的单词
算法
gugucoding6 小时前
31. 【C语言】堆栈与队列的实现
c语言·开发语言·数据结构·链表
ChaoZiLL8 小时前
我的数据结构3——链表(link list)
数据结构·链表
a1117768 小时前
LM 算法迭代过程动画演示(SLAM)
算法
头茬韭菜8 小时前
Context 的生死抉择:四层压缩、截断算法与 Session Memory
算法·ai
Jerry8 小时前
LeetCode 541. 反转字符串 II
算法
Jerry9 小时前
LeetCode 344. 反转字符串
算法