LeetCode - 114. 二叉树展开为链表 - java

LeetCode - 114. 二叉树展开为链表

---

 

题意解析

给了我们一棵二叉树【前序遍历有序】将其准换成单向链表的形式。

---

 

解题思路

因为二叉树前序遍历的结果是有序的。所以，我们可以利用一个线性表【顺序表】来存储 前序遍历的路径。之后在通过 线性表 一次读取两个节点（prev 和 cur，cur 是 prev的前序遍历的后一个节点。【在下一次读取的时候，其中一个节点是上回调用的，用于链接】）的信息，修改树节点左右孩子指针的指向即可【prev.left = null; prev.right = cur;】。

---

 

代码如下

/**
 * 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) {
        List<TreeNode> list = new ArrayList<>();
        preorderTraversal(list,root);
        int size = list.size();
        // 一次读取两个节点，修改左右孩子指针的指向（其中一个节点上回调用的，用于链接）
        for(int i = 1; i < size;i++){
            TreeNode prev = list.get(i - 1),cur = list.get(i);
            prev.left = null;
            prev.right = cur;
        }
        
    }
    // 将前序遍历的路径存储到 线性表 list 中
    private void preorderTraversal(List<TreeNode> list,TreeNode root){
        if(root==null){
            return;
        }
        list.add(root);
        preorderTraversal(list,root.left);
        preorderTraversal(list,root.right);
    }
}