LeetCode HOT 100 —— 114.二叉树展开为链表

题目

给你二叉树的根结点 root ,请你将它展开为一个单链表:

展开后的单链表应该同样使用 TreeNode ,其中 right 子指针指向链表中下一个结点,而左子指针始终为 null 。
展开后的单链表应该与二叉树 先序遍历 顺序相同。

在这里插入图片描述

思路

将二叉树展开为单链表之后,单链表中的节点顺序即为二叉树的前序遍历访问各节点的顺序。

因此,可以对二叉树进行前序遍历,获得各节点被访问到的顺序。

由于将二叉树展开为链表之后会破坏二叉树的结构,因此在前序遍历结束之后更新每个节点的左右子节点的信息(比如不要忘了左结点置空),将二叉树展开为单链表。

前序遍历可以通过递归或者迭代的方式实现

一:递归法(调用系统的栈)

java代码如下:

class Solution{
	public void flatten(TreeNode root){
		List<TreeNode> list = new ArrayList<>();
		preorderTraversal(root,list);//前序遍历生成的列表
		int size = list.size();//获取列表长度
		for (int i = 1; i < size; i++) {
			TreeNode pre = list.get(i-1);//取前驱节点
			TreeNode cur = list.get(i); //取当前节点
			pre.left = null;//左子树置空
			pre.right = cur;//右子树指向当前节点,实现链表的单向拼接
		}
	}
		
	public void preorderTraversal(TreeNode root, List<TreeNode> list) {//前序遍历:中→左→右
		if (root != null) {
			list.add(root);//中   处理逻辑:添加节点 
			preorderTraversal(root.left, list);//左
			preorderTraversal(root.right, list);//右
		}
	}
}

二、迭代法(手动维护一个栈)

java代码如下:

class Solution {
	public void flatten(TreeNode root){
		List<TreeNode> list = new ArrayList<>();//用来存储前序遍历结果
		Stack<TreeNode> stack = new Stack<>();//用来前序遍历二叉树
		TreeNode node = root;
		while(node != null || !stack.isEmpty()){
			while(node != null){//找到最左下的节点为止(即找到节点没有左孩子为止)
				list.add(node);//加入节点到结果集中
				stack.push(node);//节点入栈
				node = node.left;
			}
			node = stack.pop();//出左下的节点
			node = node.right;//继续查找左下节点的右孩子
		}
		//经过上面的循环之后,全部遍历所有节点
		int size = list.size();
		for(int i = 1; i < size; i++){
			TreeNode pre = list.get(i-1);
			TreeNode cur = list.get(i);
			pre.left = null;//左子树置空
			pre.right = cur;//右子树指向当前节点
		}
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

HDU-五七小卡

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值