[LeetCode]Flatten Binary Tree to Linked List

最新推荐文章于 2022-09-06 16:04:14 发布

原创最新推荐文章于 2022-09-06 16:04:14 发布 · 511 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

102 篇文章

订阅专栏

本文介绍了一种将二叉树转换为链表的方法。通过递归方式处理根节点、左子树和右子树，确保转换过程中的链表结构正确形成。最终实现了二叉树的扁平化。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

Given a binary tree, flatten it to a linked list in-place.

For example,
Given

The flattened tree should look like:

将二叉树变为链表结构。

解题思路

将二叉树分为三部分考虑，root，leftTree，rightTree，按如下步骤进行递归：

将右子树flatten，并得到其最末尾的节点；
将左子树flatten，并得到其最末尾的节点；
将左子树的尾节点的右指针指向右子树的根节点；
root的右指针指向左子树的根节点；

代码

        /**
	 * 将二叉树变为链表结构
	 * @param root
	 */
	public static void flatten(TreeNode root) {
		getFlattenTree(root);
	}

	/**
	 * 将二叉树变为链表结构,并返回其尾节点
	 * @param root
	 * @return
	 */
	public static TreeNode getFlattenTree(TreeNode root) {
		if(root==null)
			return null;
		if (root.left == null && root.right == null)
			return root;

		TreeNode leftTailNode = null;
		TreeNode rightTailNode = null;

		//如果右子树不为空，得到右子树flatten后的最末节点
		if (root.right != null) {
			rightTailNode = getFlattenTree(root.right);
		}
		
		if (root.left != null) {
			//如果右子树不为空，得到左子树flatten后的最末节点
			leftTailNode = getFlattenTree(root.left);
			if(root.right == null){
				//如果右子树为空，则最后返回的是左子树flatten后的最末节点
				rightTailNode = leftTailNode;
			}
			//将左子树插入到右子树之前
			leftTailNode.right = root.right;
			leftTailNode.left = null;
			root.right = root.left;
			//左子树赋为空
			root.left = null;
		}

		//返回flatten后的尾节点
		return rightTailNode;
	}