102. Binary Tree Level Order Traversal

题意:

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

代码

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;
}
全部评论

相关推荐

04-01 16:02
已编辑
武汉工程大学 Java
沉淀小子:不太懂你强调第一次面的意思,感觉没必要强调,有面试就去面,少搞点焦虑
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务