美文网首页
算法进阶——按之字形顺序打印二叉树

算法进阶——按之字形顺序打印二叉树

作者: 拉普拉斯妖kk | 来源:发表于2024-01-08 09:52 被阅读0次

题目


给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)。

数据范围:0≤n≤1500,树上每个节点的val满足∣val∣<=1500

要求:空间复杂度:O(n),时间复杂度:O(n)

例如:
给定的二叉树是{1,2,3,#,#,4,5}

该二叉树之字形层序遍历的结果是
[
[1],
[3,2],
[4,5]
]

示例1

输入:
{1,2,3,#,#,4,5}
返回值:
[[1],[3,2],[4,5]]
说明:
如题面解释,第一层是根节点,从左到右打印结果,第二层从右到左,第三层从左到右。

示例2

输入:
{8,6,10,5,7,9,11}
返回值:
[[8],[10,6],[5,7,9,11]]

示例3

输入:
{1,2,3,4,5}
返回值:
[[1],[3,2],[4,5]]

思路


用两个辅助栈,通过层序遍历二叉树,需要注意的是因为奇数层要从左往右打印,所以前一个偶数层入栈需要先右后左,偶数层正好相反。

总结规律就是奇数层先左后右入栈,偶数层先右后左入栈。

解答代码


/**
 * struct TreeNode {
 *  int val;
 *  struct TreeNode *left;
 *  struct TreeNode *right;
 *  TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
#include <stack>
#include <vector>
class Solution {
public:
    /**
     * @param pRoot TreeNode类 
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > Print(TreeNode* pRoot) {
        // write code here
        vector<vector<int> > res{};
        if (pRoot == nullptr) {
            return res;
        }

        stack<TreeNode*> odd_level;// 奇数层,从左往右打印
        stack<TreeNode*> even_level;// 偶数层,从右往左打印
        int cur = 1;// 当前层索引
        odd_level.push(pRoot);

        vector<int> tmp;
        while (!odd_level.empty() || !even_level.empty()) {
            if ((cur&1) == 1) {
                // 奇数层
                auto size = odd_level.size();
                for (int i = 0; i < size; i++) {
                    auto node = odd_level.top();
                    tmp.push_back(node->val);

                    // 先左后右
                    if (node->left != nullptr) {
                        even_level.push(node->left);
                    }
                    if (node->right != nullptr) {
                        even_level.push(node->right);
                    }

                    odd_level.pop();
                }
            } else {
                // 偶数层
                auto size = even_level.size();
                for (int i = 0; i < size; i++) {
                    auto node = even_level.top();
                    tmp.push_back(node->val);

                    // 先右后左
                    if (node->right != nullptr) {
                        odd_level.push(node->right);
                    }
                    if (node->left != nullptr) {
                        odd_level.push(node->left);
                    }

                    even_level.pop();
                }
            }
            ++cur;
            res.push_back(tmp);
            tmp.clear();
        }

        return res;
    }
};

相关文章

  • JZ-059-按之字形顺序打印二叉树

    按之字形顺序打印二叉树 题目描述 请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从...

  • 打印二叉树

    按之字形顺序打印二叉树 请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺...

  • 剑指offer | 按之字形顺序打印二叉树

    按之字形顺序打印二叉树 请实现一个函数按照之字顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的...

  • 二叉树的遍历

    从上往下打印二叉树 从上往下打印出二叉树的每个节点,同层节点从左至右打印。 按之字形顺序打印二叉树 请实现一个函数...

  • 剑指第三周

    对称的二叉树 其实就是要遍历嘛 按之字形顺序打印二叉树 同样是简单的层次遍历 把二叉树打印成多行 这个更简单了 栈...

  • 剑指offer编程题—按之字形打印二叉树

    题目描述请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按...

  • 面试题32 - III. 从上到下打印二叉树 III

    题目描述: 请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,...

  • 面试题32-3.从上到下打印二叉树3_hn

    题目描述 请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第...

  • 剑指offer 33- 之字形打印二叉树

    请实现一个函数按照之字形顺序从上向下打印二叉树。 即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第...

  • Java日记2018-06-04

    32.3 按之字形顺序打印二叉树上一题的扩展,使用辅助的队列 和栈来实现,注意一下注释里面顺序的坑 二叉搜索树的后...

网友评论

      本文标题:算法进阶——按之字形顺序打印二叉树

      本文链接:https://www.haomeiwen.com/subject/devnndtx.html