题目
给你二叉树的根结点 root
,请你将它展开为一个单链表:
- 展开后的单链表应该同样使用
TreeNode
,其中right
子指针指向链表中下一个结点,而左子指针始终为null
。 - 展开后的单链表应该与二叉树 先序遍历 顺序相同。
输入:root = [1,2,5,3,4,null,6]
输出:[1,null,2,null,3,null,4,null,5,null,6]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [0]
输出:[0]
提示:
- 树中结点数在范围
[0, 2000]
内 -100 <= Node.val <= 100
进阶:你可以使用原地算法(O(1)
额外空间)展开这棵树吗?
代码
普通版:非原地展开,AC
class Solution
{
public:
vector<int> preOrder;
void dfs(TreeNode *root)
{
if (root == nullptr)
{
return;
}
preOrder.push_back(root->val);
dfs(root->left);
dfs(root->right);
}
void flatten(TreeNode *root)
{
if (root != nullptr)
{
// preOrder.push_back(root->val);
dfs(root);
root->left = nullptr;
int length = preOrder.size();
int i = 1;
TreeNode *temp = root;
while (i < length)
{
root->right = new TreeNode(preOrder[i++]);
root = root->right;
}
root = temp;
}
}
};
进阶版:原地展开
……(看官方题解)