题目
给定一个二叉树,原地将它展开为链表。
例如,给定二叉树
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.