1、题目
Given a binary tree, flatten it to a linked list in-place.
For example,
Given
1 / \ 2 5 / \ \ 3 4 6
The flattened tree should look like:
1 \ 2 \ 3 \ 4 \ 5 \ 6
Hints:
If you notice carefully in the flattened tree, each node's right child points to the next node of a pre-order traversal
2、分析
二叉树摊直,本题是按照节点中序遍历的顺序。
有递归法和迭代法。
递归法:将左子树摊直,将右子树摊直,拼接根、摊直后的左子树、摊子后的右子树。重点在于寻找摊直后的左子树的叶子节点(即最下边的那个节点),将其与摊直后的右子树对接。
迭代法:中序遍历的思想,维护一个栈。
3、代码
#递归
<span style="font-size:18px;">/**
* Definition for binary tree
* 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;
flatten(root->left); //摊直左子树
flatten(root->right); //摊直右子树
if(!root->left) return; //无左子树,则直接返回
TreeNode *p=root->left;
while( p->right) p=p->right; //找出摊直后的左子树的叶子(最下边的一个节点)
p->right=root->right;
root->right=root->left;
root->left=nullptr; //容易忽略
}
};</span>
迭代
<span style="font-size:18px;">/**
* Definition for binary tree
* 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;
stack<TreeNode *> s;
TreeNode *p=root;
s.push(p);
while(!s.empty())
{
p=s.top();
s.pop();
if(p->right) s.push(p->right);
if(p->left) s.push(p->left);
p->left=nullptr;
if(!s.empty()) p->right=s.top();
}
}
};</span>