102. Binary Tree Level Order Traversal

原创于 2018-10-29 19:00:25 发布 · 133 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

58 篇文章

订阅专栏

本文介绍了一种树的层次遍历算法实现，通过使用结构体Node存储节点及其层级信息，借助双向链表进行节点存储，实现了对树的层次遍历，并返回各层节点值的二维向量。

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

题意：

树的层次遍历，基础数据结构题。

代码

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

struct Node {
	TreeNode *TN;
	int L;
	Node(TreeNode *t, int i) :TN(t), L(i) {};
};

vector<vector<int>> levelOrder(TreeNode* root) {
	list<Node> li;
	Node p = Node(root, 0);
	li.push_back(p);
	int lv = 0;
	vector<vector<int>> res;
	vector<int> cur;
	
	while (!li.empty()) {
		p = li.front();
		li.pop_front();
		if (p.L != lv) {
			res.push_back(cur);
			cur.clear();
			lv++;
		}
		cur.push_back(p.TN->val);
		if (p.TN->left)
			li.push_back(Node(p.TN->left, lv + 1));
		if (p.TN->right)
			li.push_back(Node(p.TN->right, lv + 1));
	}
	res.push_back(cur);
	return res;
}