[牛客题霸-研发] NC45-实现二叉树的先序、中序、后序遍历

----------------------------------------------本题链接----------------------------------------------

题目描述

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

示例

输入
{1,2,3}
 
返回值
[[1,2,3],[2,1,3],[2,3,1]]

思路

还是二叉树问题，前、中、后序遍历模板可参考[牛客经典必刷算法题] LC6-二叉树的后序遍历
很简单的问题，就是要注意一个小坑，否则会重复写很多代码
如果仔细思考模板，可以发现三种遍历模板框架其实差不多。
放到这个题里，我们代入模板，实际上只用一次递归就可以获得所有结果

算法过程

1. 判断异常情况
2. 递归遍历二叉树，一次性保存三种遍历结果
3. 输出结果

解答

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 the root of binary tree
     * @return int整型二维数组
     */
    private ArrayList<Integer> preArr = new ArrayList<>();
    private ArrayList<Integer> inArr = new ArrayList<>();
    private ArrayList<Integer> postArr = new ArrayList<>();
    
    public int[][] threeOrders (TreeNode root) {
    	if(root == null) return null;
        dfs(root);
        int[][] res = new int[3][preArr.size()];
        for(int i = 0; i < preArr.size(); i++){
            res[0][i] = preArr.get(i);
            res[1][i] = inArr.get(i);
            res[2][i] = postArr.get(i);
        }
        return res;
    }
    
    private void dfs(TreeNode root){
        if(root == null) return;
        preArr.add(root.val); // 前序
        dfs(root.left);
        inArr.add(root.val);  // 中序
        dfs(root.right);
        postArr.add(root.val);// 后序
    }
}