题目
https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list/
方法一:先序遍历
代码
递归实现
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
//先序遍历即可 根左右
class Solution {
public void flatten(TreeNode root) {
if(root==null)
return;
ArrayList<TreeNode> list=new ArrayList<>();
f(root,list);
for(int i=1;i<list.size();i++){
TreeNode pre=list.get(i-1);
TreeNode cur=list.get(i);
pre.left=null;
pre.right=cur;
}
}
public void f(TreeNode root,ArrayList<TreeNode> list){
if(root==null)
return;
list.add(root);
f(root.left,list);
f(root.right,list);
}
}
复杂度
时间复杂度:O(n),其中 n 是二叉树的节点数。前序遍历的时间复杂度是 O(n),前序遍历之后,需要对每个节点更新左右子节点的信息,时间复杂度也是O(n)。
空间复杂度:O(n),其中 n 是二叉树的节点数。空间复杂度取决于栈(递归调用栈或者迭代中显性使用的栈)和存储前序遍历结果的列表的大小,栈内的元素个数不会超过 n,前序遍历列表中的元素个数是 n。
结果
方法二:寻找前驱节点
分析
该节点的左子树中最后一个被访问的节点是左子树中的最右边的节点,也是该节点的前驱节点。因此,问题转化成寻找当前节点的前驱节点。
**对于当前节点,如果其左子节点不为空,则在其左子树中找到最右边的节点,作为前驱节点,将当前节点的右子节点赋给前驱节点的右子节点,然后将当前节点的左子节点赋给当前节点的右子节点,并将当前节点的左子节点设为空。**对当前节点处理结束后,继续处理链表中的下一个节点,直到所有节点都处理结束。
复杂度
时间复杂度:O(n),其中 n 是二叉树的节点数。展开为单链表的过程中,需要对每个节点访问一次,在寻找前驱节点的过程中,每个节点最多被额外访问一次。
空间复杂度:O(1)。
代码
class Solution {
public void flatten(TreeNode root) {
TreeNode cur=root;
while(cur!=null){
if(cur.left!=null){
TreeNode next=cur.left;
TreeNode pre=next;
while(pre.right!=null){
pre=pre.right;
}
pre.right=cur.right;
cur.left=null;
cur.right=next;
}
cur=cur.right;
}
}
}
结果
方法三
分析
在提交里面找到的用时最短的代码:先将左右子树分别转为链表,然后让根节点的右边连接左子树生成的链表,在这之后连接右子树生成的链表。【妙】
代码
class Solution {
public void flatten(TreeNode root) {
if(root==null)
return;
flatten(root.left);
flatten(root.right);
TreeNode tmp=root.right;
root.right=root.left;
root.left=null;
while(root.right!=null){
root=root.right;
}
root.right=tmp;
}
}