5-29 猿辅导面经

算法题1： 二叉树的层次遍历zigzag形，请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

其要求输出的样式是：[[1], [3,2], [4,5,6,7]]。和上段中的题目相比，这道题不仅要求按序输出节点值，还要求包含以下信息： 

1. 每一层所包含的树节点； 

2. 偶数层的树节点需倒序

我们同样可以利用列表存储节点，但不同的是列表每次只存储一层的节点，在遍历当前层节点的同时，存储下一层的节点，以此类推。面对要求的偶数层倒序，亦有两种解题思路，第一种是：获取到所有节点的值后，将偶数层的节点值倒序。第二种则是在获取节点的值的时候就倒序存入。以下给出两种方式的Python代码：

方式1： 
1. 按序获取每一层节点的值； 
2. 将偶数层节点的值倒序。

def Print(self, pRoot):
    resultArray = []
    if not pRoot:
        return resultArray
    curLayerNodes = [pRoot]
    isEvenLayer = True
    while curLayerNodes:
        curLayerValues = []
        nextLayerNodes = []
        isEvenLayer = not isEvenLayer
        for node in curLayerNodes:
            curLayerValues.append(node.val)
            if node.left:
                nextLayerNodes.append(node.left)
            if node.right:
                nextLayerNodes.append(node.right)
        curLayerNodes = nextLayerNodes
        resultArray.append(curLayerValues[::-1]) if isEvenLayer else resultArray.append(curLayerValues)
    return resultArray

方式2： 
获取每一层的节点的值时，如果是偶数层，则将每个节点的值插入到列表的头部，即实现了获取节点值时如果是偶数层则倒序排列的效果：

def Print2(self, pRoot):
    resultArray = []
    if not pRoot:
        return resultArray
    curLayerNodes = [pRoot]
    isEvenLayer = True
    while curLayerNodes:
        curLayerValues = []
        nextLayerNodes = []
        isEvenLayer = not isEvenLayer
        for node in curLayerNodes:
            curLayerValues.insert(0, node.val) if isEvenLayer else curLayerValues.append(node.val)
            if node.left:
                nextLayerNodes.append(node.left)
            if node.right:
                nextLayerNodes.append(node.right)
        curLayerNodes = nextLayerNodes
        resultArray.append(curLayerValues)
    return resultArray

思路：

层次遍历的变种

solution：

# -*- coding:utf-8 -*-  
# class TreeNode:  
#     def __init__(self, x):  
#         self.val = x  
#         self.left = None  
#         self.right = None  
class Solution:    
    def Print(self, pRoot):    
        if not pRoot:    
            return []    
        nodeStack = [pRoot]    
        result = []    
        times = 0  
        while nodeStack:  
            times += 1  
            res = []    
            nextStack = []    
            for i in nodeStack:    
                res.append(i.val)    
                if i.left:    
                    nextStack.append(i.left)    
                if i.right:    
                    nextStack.append(i.right)    
            nodeStack = nextStack  
            if times % 2 == 0:  
                res = res[::-1]         
            result.append(res)    
        return result    

相关链接：

顺序打印二叉树节点：初级版本

顺序打印二叉树节点：中级版本