leetcode_114.二叉树展开为链表(C++,图解非递归原地算法)

题目

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

例如,给定二叉树

    1
   / \
  2   5
 / \   \
3   4   6

将其展开为:

1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6

思路

我们取根节点作递归使用,每次对根节点进行操作,然后根节点指向它的右孩子,操作如下:

  • 如果左子树不为空, 那么就先递归找到左子树的最右节点(就是一直向右边找……)
  • 然后将根的右子树放到刚刚找到的最右节点的右子树上
  • 这时候根的右孩子可以释放,
  • 因此我令左孩子放到右孩子上 将左孩子置为空 继续下一个节点

怎么样,看蒙了没?(不出我所料

接下来是图解:
NO.1

NO.2

NO.3

NO.4

NO.5

NO.6

这就是整个牛逼哄哄的原地In_Place算法的实现过程。

至于每个节点的数值,你可以自己带入进去验算一下,我用人品担保,不会出错的。

以下是代码(你也可以将代码与上面的图片参照着看):

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void flatten(TreeNode* root) {
        while(root != nullptr) {
            if (root -> left != nullptr) {
                auto mr = root -> left; //如果左儿子非空,找到最右节点mr
                while(mr -> right != nullptr) //循环查找最右节点
                    mr = mr -> right;
                mr -> right = root -> right; //将根节点的右子树移到最右节点右边
                root -> right = root -> left; //将根节点的左子树移到右子树
                root -> left = nullptr; //砍掉左子树
            }
            root = root -> right; //root向右移动
        }
        return; //不管是不是void,返回都是一个好习惯
    }
};

兄台,我画图画得这么幸苦,点个赞不过分吧?

I am very vegetable.

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值