LeetCode 剑指 Offer 32 - III. 从上到下打印二叉树 III -- 利用二叉树层次遍历求解

异已

已于 2022-04-07 17:59:16 修改

阅读量57

点赞数

分类专栏： LeetCode 文章标签：算法 c++ leetcode

于 2022-04-07 17:58:15 首次发布

本文链接：https://blog.csdn.net/yllblog/article/details/124022593

版权

LeetCode 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

题目描述：请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。

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

返回其层次遍历结果：

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

代码详解:

/**
 * 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>> levelOrder(TreeNode* root) {
        vector<vector<int> > result;
        if(! root)
            return result; 

        queue<TreeNode*> q;
        q.push(root);
        result.push_back(vector<int>());
        int level = 0;
        TreeNode* last = root, * rear = root;
        while (!q.empty())
        {
            TreeNode* node = q.front();
            q.pop();
            result[level].push_back(node->val);
            if(node->left){
                q.push(node->left);
                rear = node->left;
            }
            if(node->right){
                q.push(node->right);
                rear = node->right;
            }
            
            if(node == last){
                //翻转元素
                if(level % 2 == 1)
                    reverse(result[level].begin(), result[level].end());
                
                if(rear != last){
                    ++level;
                    result.push_back(vector<int>());
                    last = rear;
                }
            }
            
        }

        return result;
    }
};