leetcode——二叉树展开为链表(java)
给你二叉树的根结点 root
,请你将它展开为一个单链表:
-
展开后的单链表应该同样使用
TreeNode
,其中right
子指针指向链表中下一个结点,而左子指针始终为null
。 -
展开后的单链表应该与二叉树 先序遍历 顺序相同。
示例 1:
输入: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]
解题方法:(头插法)(自底向上)
1.经过对题目的分析得出这道题可以使用头插法来进行解题。
2.首先判断当前的节点是否为非空节点,如果是则直接返回,否则继续。
-
首先我们先进入右子树的递归。
-
然后我们进行递归左子树
-
记得将左节点赋值为空。
-
root为链表的头节点,需要更新head为root。
总结:
-
采用逆前序遍历(右 → 左 → 根),保证转换顺序符合前序遍历(根 → 左 → 右)。
-
head
变量始终存储已经展平的链表头部,每次将当前节点插入链表头部,形成新链表。 -
最终
left
指针全部置空,right
指针指向前序遍历顺序的下一个节点,实现展平。
/**
* 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 {
private TreeNode head;
public void flatten(TreeNode root) {
if (root == null) {
return;
}
flatten(root.right);
flatten(root.left);
root.left = null;
root.right = head;
head = root;
}
}