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

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

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 {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param root TreeNode类 the root of binary tree
     * @return int整型二维数组
     */
    public int[][] threeOrders (TreeNode root) {
        // write code here
        ArrayList<Integer> arr1 = new ArrayList<>();
        ArrayList<Integer> arr2 = new ArrayList<>();
        ArrayList<Integer> arr3 = new ArrayList<>();

        if (root != null) {
            first(root, arr1);
            mid(root, arr2);
            behind(root, arr3);
        }

        int[] res1 = new int[arr1.size()];
        for (int i = 0; i < arr1.size(); i++) {
            res1[i] = arr1.get(i);
        }
        int[] res2 = new int[arr2.size()];
        for (int i = 0; i < arr2.size(); i++) {
            res2[i] = arr2.get(i);
        }
        int[] res3 = new int[arr3.size()];
        for (int i = 0; i < arr3.size(); i++) {
            res3[i] = arr3.get(i);
        }

        return new int[][] {res1, res2, res3};
    }

    public void first(TreeNode root, ArrayList<Integer> arr) {
        if (root != null) {
            arr.add(root.val);
            first(root.left, arr);
            first(root.right, arr);
        }
    }

    public void mid(TreeNode root, ArrayList<Integer> arr) {
        if (root != null) {
            mid(root.left, arr);
            arr.add(root.val);
            mid(root.right, arr);
        }
    }

    public void behind(TreeNode root, ArrayList<Integer> arr) {
        if (root != null) {
            behind(root.left, arr);
            behind(root.right, arr);
            arr.add(root.val);
        }
    }
}

全部评论

相关推荐

01-11 08:47
门头沟学院 Java
choumoduji...:读研的目的就是为了以最快的速度和最低的要求完成“学校”规定的毕业标准,而不是所谓课题组的要求
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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