题目描述:https://leetcode.com/problems/flatten-binary-tree-to-linked-list/
这个题的关键是找到左右子树的最右下角的位置,因为这将是下一次接子树的起点。下面这个方法用了递归,假设左右子树将在helper中接好,并且helper返回的是接好的子树的最右下角的位置。
class Solution {
public:
void flatten(TreeNode* root) {
helper(root);
}
TreeNode* helper(TreeNode* root){
if (root==NULL) return NULL;
TreeNode* left=helper(root->left);// left记录了root->left的最右下角的位置,用于接上root->right
TreeNode* right=helper(root->right);//right 记录了root->right最右下角的位置,拼接完成后这将是root的最右下角
if (left!=NULL)
{
left->right=root->right;//将右子树接到左子树最下面
root->right=root->left;//将整棵左子树接到右子树的位置
root->left=NULL;
}
if (right!=NULL) return right;//右子树存在最右下角的位置则优先返回这个位置(因为这个位置将是最右的)
if (left!=NULL) return left;//如果没有右子树的右下角,则左子树的右下角是最后一个位置
return root;//都没有就返回根
}
};
更好的做法:
只有左边还有子树,就用一个指针遍历到左边子树的最最右边到底的一个节点,然后把root右边子树全部挂在这个节点上,然后再通过root->right = root -> left; root -> left = NULL; 消灭右边子树,然后root = root-> right;循环往复,不断消灭右边子树。
class Solution {
public:
void flatten(TreeNode *root) {
if (root == NULL) return;
while (root) {
if (root->left) {
TreeNode *pre = root->left;
while (pre->right)
pre = pre->right;
pre->right = root->right;
root->right = root->left;
root->left = NULL;
}
root = root->right;
}
}
};