题解 | #实现二叉树先序,中序和后序遍历# Stream

实现二叉树先序,中序和后序遍历

https://www.nowcoder.com/practice/a9fec6c46a684ad5a3abd4e365a9d362

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    public List<Integer> preTree(TreeNode root) {
        if (root == null) return new ArrayList<Integer>();
        List<Integer> treeList = new ArrayList<>();
        treeList.add(root.val);
        treeList.addAll(preTree(root.left));
        treeList.addAll(preTree(root.right));
        return treeList;
    }
    public List<Integer> midTree(TreeNode root) {
        if (root == null) return new ArrayList<Integer>();
        List<Integer> treeList = new ArrayList<>();
        treeList.addAll(midTree(root.left));
        treeList.add(root.val);
        treeList.addAll(midTree(root.right));
        return treeList;
    }

    public List<Integer> lastTree(TreeNode root) {
        if (root == null) return new ArrayList<Integer>();
        List<Integer> treeList = new ArrayList<>();
        treeList.addAll(lastTree(root.left));

        treeList.addAll(lastTree(root.right));
        treeList.add(root.val);
        return treeList;
    }
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param root TreeNode类 the root of binary tree
     * @return int整型二维数组
     */
    public int[][] threeOrders (TreeNode root) {
        // write code here
        int result[][] = new int[3][];
        List<Integer> pre = preTree(root);
        int preArr[] = new int[pre.size()];
        preArr = pre.stream().mapToInt(Integer::valueOf).toArray();
        result[0] = preArr;
        List<Integer> mid = midTree(root);
        int midArr[] = new int[mid.size()];
        midArr = mid.stream().mapToInt(Integer::valueOf).toArray();
        result[1] = midArr;
        List<Integer> last = lastTree(root);
        int lastArr[] = new int[last.size()];
        lastArr = last.stream().mapToInt(Integer::valueOf).toArray();
        result[2] = lastArr;
        return result;
    }
}

尽可能利用 Java 提供的标准数据结构,减少自己的代码实现复杂度,不需要自己控制遍历下标。

全部评论

相关推荐

Southyeung:我说一下我的看法(有冒犯实属抱歉):(1)简历不太美观,给我一种看都不想看的感觉,感觉字体还是排版问题;(2)numpy就一个基础包,机器学习算法是什么鬼?我感觉你把svm那些写上去都要好一点。(2)课程不要写,没人看,换成获奖经历;(3)项目太少了,至少2-3个,是在不行把网上学习的也写上去。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务