自学内容网 自学内容网

leetcode——二叉树展开为链表(java)

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

  • 展开后的单链表应该同样使用 TreeNode ,其中 right 子指针指向链表中下一个结点,而左子指针始终为 null

  • 展开后的单链表应该与二叉树 先序遍历 顺序相同。

示例 1:

img

输入: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;
    }
}


原文地址:https://blog.csdn.net/W_L_MM/article/details/145425091

免责声明:本站文章内容转载自网络资源,如侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!