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

133 篇文章 0 订阅

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

例如,给定二叉树

    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;
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值