LintCode 将二叉树拆成链表

1.描述

将一棵二叉树按照前序遍历拆解成为一个假链表。所谓的假链表是说，用二叉树的 right 指针，来表示链表中的 next 指针。

注意事项

不要忘记将左儿子标记为 null，否则你可能会得到空间溢出或是时间溢出。

哪家公司问你的这个题？ Airbnb Alibaba Amazon Apple Baidu Bloomberg Cisco Dropbox Ebay Facebook Google Hulu Intel Linkedin Microsoft NetEase Nvidia Oracle Pinterest Snapchat Tencent Twitter Uber Xiaomi Yahoo Yelp Zenefits

感谢您的反馈

样例

              1
               \
     1          2
    / \          \
   2   5    =>    3
  / \   \          \
 3   4   6          4
                     \
                      5
                       \
                        6

2.分析

这里我们要得到的是一个假链表，其本质还是一棵树。只是这棵树是一颗斜树，左孩子都为NULL，

通过右孩子把原链表的节点值串起来，这里原二叉树的节点值是通过前序遍历的。

3.代码

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param root: a TreeNode, the root of the binary tree
     * @return: nothing
     */
    vector<int> v;
    void dfs(TreeNode *root)//前序遍历原二叉树
    {
        if(root==NULL)
        return ;
        v.push_back(root->val);
        dfs(root->left);
        dfs(root->right);
    }
    void flatten(TreeNode *root) {
        // write your code here
        dfs(root);//把原二叉树的节点值按照前序遍历存入vector中
        TreeNode *dummy;
        dummy=root;
        for(int i=1;i<v.size();i++)//尾插法
        {
            TreeNode *rear=new TreeNode(v[i]);
            dummy->left=NULL;//左子树为空
            dummy->right=rear;//右子树为下一个节点
            dummy=rear;//当前节点为新的尾节点
        }
    }
};

4.总结

通过前序遍历每得到一个节点值新建一个节点作为斜树当前节点的右孩子，同时令左孩子为空，得到斜树

即为假链表。