Binary Tree Level Order Traversal 二叉树的层次遍历

给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果:

[
  [3],
  [9,20],
  [15,7]
]

思路

二叉树的层序遍历是基础,这里参考之前一道题的discuss部分,用一个queue即可实现,不用两个queue(每次从一个出去填充另外一个),通过判断每次queue的size达到只在内部循环size次来避免访问到子节点。

代码如下:

    vector<vector<int>> levelOrder(TreeNode* root) {
	vector<vector<int>> res;
	if (!root) {
		return res;
	}
	queue<TreeNode*> queue;
	queue.push(root);
	while (!queue.empty()) {
		int q = queue.size();
		vector<int> tmp;
		for (int i = 0; i < q; i++) {
			tmp.push_back(queue.front()->val);
			if (queue.front()->left) {
				queue.push(queue.front()->left);
			}
			if (queue.front()->right) {
				queue.push(queue.front()->right);
			}
			queue.pop();
		}
		res.push_back(tmp);
	}
	return res;        
    }





猜你喜欢

转载自blog.csdn.net/qq_26410101/article/details/80538461
今日推荐