leetcode Binary Tree Zigzag Level Order Traversal

最新推荐文章于 2021-02-05 11:10:39 发布

shashafbqc

最新推荐文章于 2021-02-05 11:10:39 发布

阅读量300

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/shashafbqc/article/details/25741739

leetcode 专栏收录该内容

30 篇文章

订阅专栏

本文介绍了一种使用两个栈实现的二叉树节点值的锯齿形层序遍历算法。该算法从根节点开始，通过交替改变层次遍历的方向（先从左到右再从右到左），返回每层节点值的列表。

摘要生成于 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) {}
 * };
 */
/*这个题目我还是采用了一直使用的两个stack（如果是层次的话就用两个queue）来处理，比较直观。需要注意的是，当处理下一层是，要分层次的奇偶，来判断是先压入左孩子，还是先压入右孩子*/
class Solution {
    
public:
    vector<vector<int> > zigzagLevelOrder(TreeNode *root) { 
            vector<vector<int>> re;
        if(root==NULL) return re;
        stack<TreeNode *> q;
        stack<TreeNode *> s;
		q.push(root);
		while(!q.empty()||!s.empty()){
			vector<int> temp;
			if(!q.empty()){
				while(!q.empty()){
					TreeNode * cur=q.top();
					temp.push_back(cur->val);
					if(cur->left!=NULL) s.push(cur->left);
					if(cur->right!=NULL) s.push(cur->right);
					q.pop();
				}
				re.push_back(temp);
				continue;
			}
			if(!s.empty()){
				while(!s.empty()){
					TreeNode * cur=s.top();
					temp.push_back(cur->val);
					if(cur->right!=NULL) q.push(cur->right);
					if(cur->left!=NULL) q.push(cur->left);
					
					s.pop();
				}
				re.push_back(temp);
			}
		}
        return re;
        
    }
};