【leetcode 二叉树摊直】Flatten Binary Tree to Linked List

1、题目

Given a binary tree, flatten it to a linked list in-place.

For example,
Given

         1
        / \
       2   5
      / \   \
     3   4   6

The flattened tree should look like:

   1
    \
     2
      \
       3
        \
         4
          \
           5
            \
             6

Hints:

If you notice carefully in the flattened tree, each node's right child points to the next node of a pre-order traversal

2、分析

二叉树摊直，本题是按照节点中序遍历的顺序。

有递归法和迭代法。

递归法：将左子树摊直，将右子树摊直，拼接根、摊直后的左子树、摊子后的右子树。重点在于寻找摊直后的左子树的叶子节点（即最下边的那个节点），将其与摊直后的右子树对接。

迭代法：中序遍历的思想，维护一个栈。

【leetcode 中序遍历】Binary Tree Inorder Traversal

3、代码

#递归

<span style="font-size:18px;">/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
 /*
 将二叉树转化为链表
 按节点先序遍历的顺序
 */
class Solution {
public:
    void flatten(TreeNode *root) {
        if(!root) return;
        flatten(root->left);         //摊直左子树
        flatten(root->right);        //摊直右子树
        
        if(!root->left)  return;    //无左子树，则直接返回
        TreeNode *p=root->left; 
        while( p->right) p=p->right;  //找出摊直后的左子树的叶子（最下边的一个节点）
        
        p->right=root->right;      
        root->right=root->left;  
        root->left=nullptr;  //容易忽略
    }
};</span>

迭代

<span style="font-size:18px;">/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
 /*
 将二叉树转化为链表
 按节点先序遍历的顺序
 */
class Solution {
public:
    void flatten(TreeNode *root) {
        if(!root) return;
        stack<TreeNode *> s;
        TreeNode *p=root;
        s.push(p);
        while(!s.empty())
        {
            p=s.top();
            s.pop();
            
            if(p->right) s.push(p->right);
            if(p->left)  s.push(p->left);
            
            p->left=nullptr;
           if(!s.empty()) p->right=s.top();
        }
    }
};</span>