【leetcode刷题】T116-二叉树的锯齿形层次遍历

木又AI帮

发布于 2019-07-17 22:04:25

35200

代码可运行

文章被收录于专栏：木又AI帮木又AI帮

运行总次数：0

代码可运行

木又连续日更第68天（68/100）

木又的第116篇leetcode解题报告

二叉树类型第6篇解题报告

leetcode第103题：二叉树的锯齿形层次遍历

https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal/

【题目】

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

例如：
给定二叉树 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回锯齿形层次遍历如下：

[
  [3],
  [20,9],
  [15,7]
]

【思路】

按照层次遍历的方法，当某一层需要从右往左遍历时，将数组num反转。

【代码】

python版本

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def zigzagLevelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        if not root:
            return []
        res = []
        num = []
        tmp1 = [root]
        tmp2 = []
        reverse = False
        while len(tmp1) != 0 or len(tmp2) != 0:
            if len(tmp1) == 0:
                tmp1 = copy.copy(tmp2)
                tmp2 = []
                if reverse:
                    reverse = False
                    num = num[::-1]
                else:
                    reverse = True
                res.append(num)
                num = []
            p = tmp1.pop(0)
            num.append(p.val)
            if p.left:
                tmp2.append(p.left)
            if p.right:
                tmp2.append(p.right)
        if reverse:
            num = num[::-1]
        res.append(num)
        return res

C++版本

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        vector<vector<int>> res;
        if(!root)
            return res;
        vector<int> num;
        vector<TreeNode*> tmp1;
        vector<TreeNode*> tmp2;
        tmp1.push_back(root);
        TreeNode* p = NULL;
        bool should_reverse = false;
        while(tmp1.size() != 0 || tmp2.size() != 0){
            // tmp1为空，tmp2非空
            if(tmp1.size() == 0){
                for(int i=0; i<tmp2.size(); i++){
                    tmp1.push_back(tmp2[i]);
                }
                tmp2.erase(tmp2.begin(), tmp2.end());
                if(should_reverse){
                    reverse(num.begin(), num.end());
                    should_reverse = false;
                }else{
                    should_reverse = true;
                }
                res.push_back(num);
                num.erase(num.begin(), num.end());
            }
            p = tmp1[0];
            num.push_back(p->val);
            tmp1.erase(tmp1.begin());
            if(p->left)
                tmp2.push_back(p->left);
            if(p->right)
                tmp2.push_back(p->right);
        }
        if(should_reverse){
            reverse(num.begin(), num.end());
        }
        res.push_back(num);
        return res;
    }
};

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2019-07-14，如有侵权请联系 cloudcommunity@tencent.com 删除

二叉树