Leetcode 二叉树的锯齿形层序遍历

二叉树的锯齿形层序遍历

题目描述：

给定一个二叉树，返回其节点值的锯齿形层序遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。

题目链接

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    private List<List<Integer>> result = new ArrayList<List<Integer>>();
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        helper(root,1);
        return this.result;
    }
    // 第一个参数代表当前需要层序遍历的根节点，第二个参数代表当前的节点层数
    private void helper(TreeNode root,int layers){
        if(root == null){
            return;
        }
        // 采用前序遍历
        if(layers > result.size()){
            ArrayList<Integer> temp = new ArrayList<>();
            temp.add(root.val);
            result.add(temp);
        }else{
            if(layers%2 == 0){
                result.get(layers-1).add(root.val); // 后向添加
            }else{
                result.get(layers-1).add(0,root.val); // 前向添加
            }
        }
        helper(root.right,layers+1);
        helper(root.left,layers+1);
    }
}

按照题意即可，这里采用递归方式实现，需要明确该题是前序遍历，由于锯齿形层序遍历，故可以通过对layers取2模来实现达到元素前后顺序的问题。详细请看代码，有疑问欢迎留言。