题目链接:
思路:
我是先看的其他人的题解,所以就用他们的题解了。
拿到一个二叉树的题还是考虑“操作+遍历”的思路。通过看图可以发现,操作就是把右子树转到左子树上,然后循环这个过程直到回到了根节点那里为止。那么这个操作只需要几个额外的TreeNode*就能解决,类似于不同链表之间的遍历和插入,所以不需要新加一个操作函数,在本函数内部就能解决。因为要找到右子树对右子树进行操作,所以应该是后序遍历,操作放在后序遍历之后。
那么假设现在已经到了右子树了,该如何进行操作呢?根据题意,操作完后原来root的左子节点是空,而右子节点指向的是左子节点的位置,所以先进行一个嫁接操作,这一步是让右节点指向原来左边的那一串树;之后就需要把右边的分支补到左边的末尾上了,这就相当于一个链表寻尾再尾插的操作了。
如果不太明白,可以用一个最简单的两层二叉树去模拟一下操作的过程,这个明白之后,其余的大支都是这么做的,就能理解了。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
void flatten(TreeNode* root)
{
if(!root)
{
return;
}
flatten(root->left);
flatten(root->right);
TreeNode* l=root->left;
TreeNode* r=root->right;
root->left=NULL;
root->right=l;
TreeNode* p=root;
while(p->right)
{
p=p->right;
}
// p=p->right;
p->right=r;
}
};