leetcode Binary Tree Zigzag Level Order Traversal

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

原创最新推荐文章于 2021-02-05 11:10:39 发布 · 306 阅读

0 ·

CC 4.0 BY-SA版权

leetcode Fighting 专栏收录该内容

133 篇文章

订阅专栏

本文介绍了一种实现二叉树锯齿形层序遍历的方法，通过两个队列交替存储节点，利用布尔标志判断是否需要反转当前层的节点值顺序，最终返回按锯齿形排列的每层节点值。

代码

class Solution
{
public:
	vector<vector<int> > zigzagLevelOrder(TreeNode *root) 
	{
		vector<vector<int> > result;

		if(root==NULL)
			return result;

		queue<TreeNode*> queue_pop, queue_push;
		queue_pop.push(root);

		vector<int> level;
        bool flag = false;
		while(!queue_pop.empty())
		{
			TreeNode *tempNode;
			tempNode = queue_pop.front();
			level.push_back(tempNode->val);
			queue_pop.pop();
			

			if(tempNode->left)
				queue_push.push(tempNode->left);
			if(tempNode->right)
				queue_push.push(tempNode->right);		

			if(queue_pop.empty())
			{
			    if(flag)
			    {
			        flag = false;
			        reverse(level.begin(), level.end());
			    }
			    else
			       flag = true;
				result.push_back(level);
				level.clear();	
				swap(queue_push, queue_pop);
			}
		
		}

		return result;
    }


};