[leetcode]Flatten Binary Tree to Linked List

最新推荐文章于 2022-09-06 16:04:14 发布

转载最新推荐文章于 2022-09-06 16:04:14 发布 · 49 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/lautsie/p/3267682.html

本文介绍了一种使用栈技术优化先序遍历树的方法，通过记录上一次访问的节点，简化了遍历过程，提高了效率。

先序遍历的一个变化。记录下上次的节点就行了。

public class Solution {
    public void flatten(TreeNode root) {
        Stack<TreeNode> stack = new Stack<TreeNode>();
        TreeNode n = root;
        TreeNode last = null;
        while (n != null || !stack.empty()) {
            if (n != null) {
                if (last != null) {
                    last.left = null;
                    last.right = n; 
                }
                last = n;
                if (n.right != null) {
                    stack.push(n.right);
                }
                n = n.left;
            }
            else {
                n = stack.pop();
            }
        }
    }
}

转载于:https://www.cnblogs.com/lautsie/p/3267682.html