leetcode103. 二叉树的锯齿形层次遍历

最新推荐文章于 2024-09-23 11:27:55 发布

Rookie2Master

最新推荐文章于 2024-09-23 11:27:55 发布

阅读量92

点赞数

分类专栏： leetcode 文章标签： leetcode 二叉树 stack

本文链接：https://blog.csdn.net/qq_43778308/article/details/106495949

版权

leetcode 专栏收录该内容

227 篇文章 1 订阅

订阅专栏

传送门

题目:给定一个二叉树，返回其节点值的锯齿形层次遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。
给定二叉树 [3,9,20,null,null,15,7],

返回锯齿形层次遍历如下：[[3],[20,9],[15,7]]

两个栈来回装层次遍历的节点,
并且根据栈的索引判断由左到右还是右到左的顺序

	 List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> ans = new  ArrayList<List<Integer>>();
        if (root == null) return ans;
		//不能声明泛型数组Stack<TreeNode>[] s=new…  
		//所以把栈装进list里
        List<Stack<TreeNode>> s = new ArrayList<Stack<TreeNode>>();
        s.add(new Stack<TreeNode>());// 两个栈
        s.add(new Stack<TreeNode>());
        int stackIndex = 0;//当前遍历的栈
        s.get(0).push(root);

        while (!s.get(stackIndex).empty() ||
        	!s.get(1-stackIndex).empty()) {
        	
            List<Integer> tmpList = new ArrayList<Integer>();
            while (!s.get(stackIndex).empty()) { // 遍历当前栈
                TreeNode tmpNode = s.get(stackIndex).pop();
                tmpList.add(tmpNode.val);
                // 偶数层,左到右压入奇栈s[1],右到左读取
                if (stackIndex == 0) { 
                    if (tmpNode.left != null) 
                        s.get(1).push(tmpNode.left); 
                    if (tmpNode.right != null) 
                        s.get(1).push(tmpNode.right);
                } else {        // 奇数层..相反
                    if (tmpNode.right != null) 
                        s.get(0).push(tmpNode.right);
                    if (tmpNode.left != null) 
                        s.get(0).push(tmpNode.left);
                }
            }
            stackIndex = 1 - stackIndex; // 栈的索引,0变1,1变0
            ans.add(tmpList);
        }
        return ans;
    }