给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [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; }