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

描述

答案：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 preList = new ArrayList<>();
        List inList = new ArrayList<>();
        List 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 preList) {
        if(root == null) return;
        preList.add(root.val);
        preOrder(root.left, preList);
        preOrder(root.right, preList);
    }
    
    public void inOrder(TreeNode root, List inList) {
        if(root == null) return;
        inOrder(root.left, inList);
        inList.add(root.val);
        inOrder(root.right, inList);
    }
    
    public void postOrder(TreeNode root, List postList) {
        if(root == null) return;
        postOrder(root.left, postList);
        postOrder(root.right, postList);
        postList.add(root.val);
    }
}