103. Binary Tree Zigzag Level Order Traversal

最新推荐文章于 2024-08-23 15:23:16 发布

ShenYounger

最新推荐文章于 2024-08-23 15:23:16 发布

阅读量267

点赞数

分类专栏： leetcode

leetcode 专栏收录该内容

220 篇文章

订阅专栏

本文介绍了一种使用两个栈实现的二叉树锯齿形层序遍历算法，该算法能按从左到右再从右到左交替的方式返回各层节点值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given a binary tree, return the zigzag level order traversal of its nodes' values. (ie, from left to right, then right to left for the next level and alternate between).

For example:
Given binary tree {3,9,20,#,#,15,7},

return its zigzag level order traversal as:

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

分析：首先肯定的是要对树进行层序遍历，但是相邻两层的元素遍历顺序是相反的，因此传统的非递归遍历方法用一个队列肯定是无法实现。我们用两个栈来实现，同一层元素都在同一个栈中，相邻的两层元素放在不同的栈中，比如第一层元素放在第一个栈，那么第二层就放在第二个栈，第三层就放在第三个栈......如果某一层元素是从左往右遍历的，那么这层元素的孩子节点入栈顺序就是先左孩子后右孩子，相反如果某一层元素是从右往左遍历的，入栈顺序就是先右孩子后左孩子。

/**
 * Definition for binary tree
 * 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) {
        // IMPORTANT: Please reset any member data you declared, as
        // the same Solution instance will be reused for each test case.
        vector<vector<int> >res;
        if(root == NULL)return res;
        stack<TreeNode*> S,S2;
        S.push(root);
        bool isS = true;
        vector<int>tmpres;
        while(S.empty() == false || S2.empty() == false)
        {
            TreeNode *p;
            if(isS)
            {
                
                p = S.top();
                S.pop();
                tmpres.push_back(p->val);
                if(p->left)S2.push(p->left);
                if(p->right)S2.push(p->right);
                if(S.empty() == true)
                {
                    res.push_back(tmpres);
                    tmpres.clear();
                    isS = false;
                }
            }
            else
            {
                p = S2.top();
                S2.pop();
                tmpres.push_back(p->val);
                if(p->right)S.push(p->right);
                if(p->left)S.push(p->left);
                if(S2.empty() == true)
                {
                    res.push_back(tmpres);
                    tmpres.clear();
                    isS = true;
                }
            }
        }
        return res;
    }
};