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

题目


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

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

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

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

算法进阶——按之字形顺序打印二叉树_第1张图片

该二叉树之字形层序遍历的结果是
[
[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 
#include 
class Solution {
public:
    /**
     * @param pRoot TreeNode类 
     * @return int整型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;
    }
};

你可能感兴趣的:(算法,算法)