Leetcode:114. 二叉树展开为链表

给定一个二叉树,原地将它展开为链表。

例如,给定二叉树

    1
   / \
  2   5
 / \   \
3   4   6

将其展开为:

1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6

解题思路:

最近做了较多的二叉树题目,感觉使用递归的话会变得贼简单,其实细想一下二叉树的结构,有左子树,右子树,在结构上本质和根节点包含的树并没有什么区别,处理成递归是结构所导致。对于简单的递归,我们可以考虑递归消除,这样既能保证计算速度,又能不影响代码的阅读效率;然而对于比较浮渣的递归而言,强行递归消除可能得到了一时的速度优势,但是代码不方便阅读极可能影响后续的开发。当然以上只是个人看法。

 

回归到本题的解题思路:

1. 本题的问题是将一颗树转换成链表,其实这个链表并非真正的链表,本质还是树,我在这称作链表树!只不过对于任意的节点只有右孩子,正如例子给出的效果那样。

2. 我的方法是,递归,将左子树变成链表树,根为root->left,返回叶子结点LEFT_end。将右子树变成链表树root->right,返回叶子结点RIGHT_end。

3.连接,根结点root,以及这两个树。

注意,这两个树都可能不存在!

                               

C++代码
#define hasLChild(x) (!(x->left==NULL)) #define hasRChild(x) (!(x->right==NULL)) #define hasNoChild(x) (!hasLChild(x)&&!hasRChild(x)) class Solution { public:     void flatten(TreeNode* root) {         if (root == NULL) return ;         DFS(root);     }     TreeNode* DFS(TreeNode* root) {//返回链表表尾         if (hasNoChild(root)) return root;         TreeNode* LTREE_begin = root->left, *LTREE_end = NULL, *RTREE_begin = root->right, *RTREE_end = NULL;         //获取两个链表树,如果有的话         if (hasLChild(root)) LTREE_end = DFS(root->left);         if (hasRChild(root)) RTREE_end = DFS(root->right);         //链接它们         if (hasLChild(root)) {             if (hasRChild(root)) {                 root->right = LTREE_begin;                 LTREE_end->right = RTREE_begin;             }             else {                 root->right = new TreeNode(0);                 root->right = root->left;                 RTREE_end = LTREE_end;             }             root->left = NULL;         }         return RTREE_end;     } };