102. 二叉树的层序遍历

yiyakaa

于 2023-06-20 22:46:26 发布

阅读量51

点赞数

文章标签：算法数据结构

本文链接：https://blog.youkuaiyun.com/qq_44902386/article/details/131317043

版权

该代码实现了一个层次遍历二叉树的算法，通过队列进行节点的入栈和出栈，依次访问每个层级的节点，将同一层的节点值存入一个向量中，最后将所有层的节点值向量组成结果集返回。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述
思路:

用队列完成，比较简单

struct TreeNode {
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode(int x):val(x),left(NULL),right(NULL){}
};

class Solution {
public:
	vector<vector<int>> levelOrder(TreeNode* root) {
			vector<vector<int>> result;
		
			queue<TreeNode*> que;
			if (root != NULL) {
				que.push(root);
			}
		
			while (!que.empty()) {
				int size = que.size();
				vector<int> path;
				for (int i = 0; i < size;i++) {
					TreeNode* Node = que.front();
					que.pop();
					path.push_back(Node->val);
					if (Node->left)que.push(Node->left);
					if (Node->right)que.push(Node->right);
				}
				result.push_back(path);
			}
			return result;
	}
};

int main() {
	TreeNode* A = new TreeNode(3);
	TreeNode* B = new TreeNode(9);
	TreeNode* C = new TreeNode(20);
	TreeNode* D = new TreeNode(15);
	TreeNode* E = new TreeNode(7);
	A->left = B;
	A->right = C;
	C->left = D;
	C->right = E;
	Solution ss;
	vector<vector<int>> res = ss.levelOrder(A);
	for (auto& chs : res) {
		for (auto& ch : chs) {
			cout << ch << " ";
		}
		cout << endl;
	}
	return 0;
}