【每日一题】【list转int数组】【Lambda的简化-方法引用】2022年1月15日-NC45 实现二叉树先序，中序和后序遍历...

最新推荐文章于 2023-07-25 20:17:44 发布

原创最新推荐文章于 2023-07-25 20:17:44 发布 · 144 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

本文详细介绍了一种算法，该算法能够实现对一棵二叉树进行先序、中序和后序遍历，并将遍历结果以整数数组的形式返回。文章通过具体的代码实现了这一过程，并确保了空间和时间复杂度均为O(n)。

描述

给定一棵二叉树，分别按照二叉树先序，中序和后序打印所有的节点。

数据范围：

\le n \le 10000≤n≤1000，树上每个节点的val值满足 0 \le val \le 1000≤val≤100

要求：空间复杂度

样例解释：

答案：Lambda的简化-方法引用，list转int数组

import java.util.*;
public class Solution {
    /**
     * 
     * @param root TreeNode类 the root of binary tree
     * @return int整型二维数组
     */
    
    public int[][] threeOrders (TreeNode root) {
        List<Integer> preList = new ArrayList<>();
        List<Integer> inList = new ArrayList<>();
        List<Integer> postList = new ArrayList<>();
        preOrder(root, preList);
        inOrder(root, inList);
        postOrder(root, postList);
        int[][] res = new int[3][preList.size()];
        //转为int数组，需要用mapToInt
        res[0] = preList.stream().mapToInt(Integer::valueOf).toArray();
        res[1] = inList.stream().mapToInt(Integer::valueOf).toArray();
        res[2] = postList.stream().mapToInt(Integer::valueOf).toArray();
        return res;
    }
    
    public void preOrder(TreeNode root, List<Integer> preList) {
        if(root == null) return;
        preList.add(root.val);
        preOrder(root.left, preList);
        preOrder(root.right, preList);
    }
    
    public void inOrder(TreeNode root, List<Integer> inList) {
        if(root == null) return;
        inOrder(root.left, inList);
        inList.add(root.val);
        inOrder(root.right, inList);
    }
    
    public void postOrder(TreeNode root, List<Integer> postList) {
        if(root == null) return;
        postOrder(root.left, postList);
        postOrder(root.right, postList);
        postList.add(root.val);
    }
}