Leetcode103.Binary_Tree_Zigzag_Level_Order_Traversal

最新推荐文章于 2021-05-17 20:12:50 发布

NK_ZZL

最新推荐文章于 2021-05-17 20:12:50 发布

阅读量148

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/qq_42263831/article/details/84261350

版权

LeetCode 专栏收录该内容

78 篇文章 0 订阅

订阅专栏

用一个栈来使数据反向。
时间复杂度：O(N)
C++代码：

class Solution {
public:
	vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
		if (root == nullptr)
			return {};
		int dir = 1;
		vector<vector<int>> result(1);
		queue<TreeNode*> record;
		stack<TreeNode*> reverse;
		record.push(root);
		while (!record.empty())
		{
			TreeNode* temp = record.front();
			record.pop();
			(*(result.end() - 1)).push_back(temp->val);
			if (dir)
			{
				if (temp->left != nullptr)
					reverse.push(temp->left);
				if (temp->right != nullptr)
					reverse.push(temp->right);
			}
			else
			{
				if (temp->right != nullptr)
					reverse.push(temp->right);
				if (temp->left != nullptr)
					reverse.push(temp->left);
			}
			if (record.empty())
			{
				dir = !dir;
				if (!reverse.empty())
					result.push_back({});
				while (!reverse.empty())
				{
					TreeNode* now = reverse.top();
					reverse.pop();
					record.push(now);
				}
			}
		}
		return result;
	}
};