给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
一、思路
层序遍历可以使用队列来模拟,因为这里需要体现出层次感,所以使用两个队列来循环使用。
每次入队,先将左子树的根结点放入,再将右子树的根结点放入。
C++代码:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> ans;
if (root == NULL)
return ans;
queue<TreeNode*> list1, list2;
list1.push(root);
while (!list1.empty() || !list2.empty()) {
TreeNode* node;
vector<int> level_temp;
while (!list1.empty()) {
node = list1.front();
list1.pop();
level_temp.push_back(node->val);
if (node->left)
list2.push(node->left);
if (node->right)
list2.push(node->right);
}
if(!level_temp.empty())
ans.push_back(level_temp);
level_temp.clear();
while (!list2.empty()) {
node = list2.front();
list2.pop();
level_temp.push_back(node->val);
if (node->left)
list1.push(node->left);
if (node->right)
list1.push(node->right);
}
if (!level_temp.empty())
ans.push_back(level_temp);
}
return ans;
}
};
执行效率: