Binary Tree Zigzag Level Order Traversal (LeetCode)

2 篇文章 0 订阅
1 篇文章 0 订阅

Given a binary tree, return the zigzag level order traversal of its nodes' values. (ie, from left to right, then right to left for the next level and alternate between).

For example:
Given binary tree {3,9,20,#,#,15,7},

    3
   / \
  9  20
    /  \
   15   7

return its zigzag level order traversal as:

[
  [3],
  [20,9],
  [15,7]
]
Solution 1 : bfs

public ArrayList<ArrayList<Integer>> zigzagLevelOrder(TreeNode root) {
        ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
        if(root==null)
            return list;
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        ArrayList<Integer> sub = new ArrayList<Integer>();
        queue.add(root);
        boolean leftToRight=true;
        int current = 1, next = 0;
        while(!queue.isEmpty()){
            TreeNode node = queue.remove();
            if(node.left!=null){
                queue.add(node.left);
                next++;
            }
            if(node.right!=null){
                queue.add(node.right);
                next++;
            }
            if(leftToRight)
                sub.add(node.val);
            else
                sub.add(0,node.val);
            current--;
            if(current==0){
                list.add(new ArrayList<Integer>(sub));
                sub.clear();
                current=next;
                next=0;
                leftToRight=!leftToRight;
            }
        }
        return list;
    }

Solution 2: dfs

public ArrayList<ArrayList<Integer>> zigzagLevelOrder(TreeNode root) {
        // IMPORTANT: Please reset any member data you declared, as
        // the same Solution instance will be reused for each test case.
        ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
        zigzagHelper(root, 0, list);
        return list;
    }
    
    public void zigzagHelper(TreeNode root, int height, ArrayList<ArrayList<Integer>> list){
        if(root==null)
            return;
        if(height == list.size()){
            ArrayList<Integer> sub = new ArrayList<Integer>();
            sub.add(root.val);
            list.add(sub);
        }
        else{
            ArrayList<Integer> sub = list.get(height);
            if(height%2==0)
                sub.add(root.val);
            else
                sub.add(0,root.val);
        }
        zigzagHelper(root.left,height+1,list);
        zigzagHelper(root.right,height+1,list);
        
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值