LeetCode：114. Flatten Binary Tree to Linked List（固定二叉树为链表）

文章最前： 我是Octopus，这个名字来源于我的中文名--章鱼；我热爱编程、热爱算法、热爱开源。所有源码在我的个人github ；这博客是记录我学习的点点滴滴，如果您对 Python、Java、AI、算法有兴趣，可以关注我的动态，一起学习，共同进步。

相关文章：

1. LeetCode：55. Jump Game（跳远比赛）
2. Leetcode：300. Longest Increasing Subsequence（最大增长序列）
3. LeetCode：560. Subarray Sum Equals K（找出数组中连续子串和等于k）

---

文章目录

题目描述：

java实现方式1：（利用栈的形式）

python实现方式1：

 java实现方式2：递归形式

python实现方式2：递归方式​​​​​​​

源码github地址：https://github.com/zhangyu345293721/leetcode

---

题目描述：

给定一个二叉树，将它展开为链表。

例如，给定二叉树;

    1
   / \
  2   5
 / \   \
3   4   6

The flattened tree should look like:

1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6

---

java实现方式1：（利用栈的形式）

   
    /**
     * 将二叉树改为链表
     *
     * @root 根节点
     */
    public void flatten(TreeNode root) {
        if (root == null) return;
        Stack stk = new Stack();
        stk.push(root);
        while (!stk.isEmpty()){
            TreeNode curr = stk.pop();
            if (curr.right!=null)  
                 stk.push(curr.right);
            if (curr.left!=null)  
                 stk.push(curr.left);
            if (!stk.isEmpty()) 
                 curr.right = stk.peek();
            // 它的左子树要为空
            curr.left = null;  
    }
  }

时间复杂度：O(n)

空间复杂度：O(1)

---

python实现方式1：

def flatten(root: TreeNode) -> None:
    '''
        转换二叉树
    Args:
        root: 根节点
    Returns:
        只有有节点的二叉树
    '''
    if root == None:
        return
    stack = []
    stack.append(root)
    while len(stack) > 0:
        current_node = stack.pop()
        if current_node.right != None:
            stack.append(current_node.right)
        if current_node.left != None:
            stack.append(current_node.left)
        if len(stack) > 0:
            current_node.right = stack[-1]
        current_node.left = None

时间复杂度：O(n)

空间复杂度：O(1)

---

 java实现方式2：递归形式

    
    /**
     * 将二叉树改为链表
     *
     * @root 根节点
     */
    public void flatten(TreeNode root) {
        falttenForList(root, null);
    }
    private TreeNode falttenForList(TreeNode root, TreeNode cur) {
        if (root == null)
            return cur;
        cur = falttenForList(root.right, cur);
        cur = falttenForList(root.left, cur);
        root.left = null;
        root.right = cur;
        return root;
        
    }

时间复杂度：O(n)

空间复杂度：O(n)

---

python实现方式2：递归方式

def flatten_for_list(root: TreeNode, current_node) -> TreeNode:
    '''
        循环递归树
    Args:
        root: 根节点
        current_node: 当前节点
    Returns:
        root
    '''
    if root == None:
        return current_node
    current_node = flatten_for_list(root.right, current_node)
    current_node = flatten_for_list(root.left, current_node)
    root.left = None
    root.right = current_node
    return root


def flatten2(root: TreeNode) -> None:
    '''
        转换二叉树
    Args:
        root: 根节点
    Returns:
        只有有节点的二叉树
    '''
    flatten_for_list(root, None)

时间复杂度：O(n)

空间复杂度：O(n)

 

---

源码github地址：https://github.com/zhangyu345293721/leetcode