剑指 Offer 32 - III. 从上到下打印二叉树 III（C++实现）

最新推荐文章于 2025-12-20 17:21:30 发布

原创最新推荐文章于 2025-12-20 17:21:30 发布 · 86 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #数据结构 #c++

我的剑指刷题系列专栏收录该内容

29 篇文章

订阅专栏

该代码实现了一个函数，用于按层次遍历二叉树，同时可以切换顺时针或逆时针顺序。使用双端队列存储每一层的节点值，并通过一个flag变量切换遍历方向。遍历过程中，将左右子节点依次入队，最后将当前层的节点值添加到结果中。

剑指 Offer 32 - III. 从上到下打印二叉树 IIIhttps://leetcode.cn/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof/

通过一个双端队列记录遍历每一层节点时的结果

用一个 flag 变量标志遍历该层时应该为顺时针遍历还是逆时针遍历

vector<vector<int>> levelOrder(TreeNode* root) {

	// base case
	if (root == nullptr)
	{
		return {};
	}
	
	vector<vector<int>> res;
	
	bool flag = true;
	queue<TreeNode*> q;
	q.push(root);

	while (!q.empty())
	{
		int n = q.size();
		deque<int> d;				// 用双端队列 记录每一层的遍历结果

		for (int i = 0; i < n; ++i)	// 遍历当前层的每个节点
		{
			TreeNode* cur = q.front();
			q.pop();

			if (flag)				// 顺时针遍历	
			{
				d.push_back(cur->val);
			}
			else					// 逆时针遍历
			{
				d.push_front(cur->val);
			}

			if (cur->left != nullptr)
			{
				q.push(cur->left);
			}
			if (cur->right != nullptr)
			{
				q.push(cur->right);
			}
		}

		flag = !flag;
		res.emplace_back(vector<int>(d.begin(), d.end()));
	}

	return res;
}