力扣：102. 二叉树的层序遍历

题目描述

给你二叉树的根节点 root ，返回其节点值的层序遍历 。 （即逐层地，从左到右访问所有节点）。

问题分析

层序遍历需要借助一个辅助数据结构即队列实现，队列先进先出，符合一层一层遍历逻辑。

代码实现

// 编程软件：VS2019
// 参考书籍：代码随想录
#include<iostream>
#include<vector>
#include<queue>
using namespace std;

// 二叉树的定义
struct TreeNode {
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode(int val) :val(val), left(NULL), right(NULL) {}
};

vector<vector<int>> levelOrder(TreeNode* root) {
	queue<TreeNode*> que; // 辅助队列 
	if (root != NULL) que.push(root);
	vector<vector<int>> result;
	while (!que.empty()) {
		// 这里一定要使用固定大小size，不要使用que.size()，因为que.size()是不断变化的
		int size = que.size(); // 统计每一层的节点数
		vector<int> vec;
		// 遍历这一层的节点 ，并将下一层的节点加入队列 
		for (int i = 0; i < size; i++) { 
			TreeNode* node = que.front();
			que.pop();
			vec.push_back(node->val);
			if (node->left) que.push(node->left);
			if (node->right) que.push(node->right);
		}
		result.push_back(vec);
	}
	return result;
}

int main() {
	TreeNode* root = new TreeNode(1);
	root->left = new TreeNode(2);
	root->right = new TreeNode(3);
	root->left->left = new TreeNode(4);
	root->left->right = new TreeNode(5);
	root->left->left->left = new TreeNode(6);
	vector<vector<int>> res;
	res = levelOrder(root);
	for (int i = 0; i < res.size(); i++) {
		for (int j = 0; j < res[i].size(); j++) {
			cout << res[i][j]<<" ";
		}
		cout << endl;
	}
}
/*
输出结果：
1
2 3
4 5
6 
*/