题目描述
给定一个二叉树,原地将它展开为链表。
例如,给定二叉树
1 / \ 2 5 / \ \ 3 4 6
将其展开为:
1 \ 2 \ 3 \ 4 \ 5 \ 6
问题分析
此题用递归的方法求解。先用变量保存根节点的右孩子,然后分别对根节点的左子树和右子树调用递归函数,使左右子树分别变为链表,返回后将根节点的右孩子设置为根节点的左孩子,将根节点的左孩子设置为NULL。然后用while循环一直向右寻找,直到底部的节点,然后将底部的节点的右孩子设置为之前保存的根节点的右孩子。递归基是当节点为空时返回。这样就完成了二叉树到链表的转换。
代码实现
/**
* 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) {
if(!root)
return;
TreeNode* temp = root->right;
flatten(root->left);
flatten(root->right);
root->right = root->left;
root->left = NULL;
TreeNode* cur = root;
while(cur->right)
cur = cur->right;
cur->right = temp;
}
};