难度:简单
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
提示:
节点总数 <= 1000
注意:本题与 102. 二叉树的层序遍历 相同。
这里借助 优先队列 来实现 广度优先遍历:
cnt
每访问一层,都要先记录此时队列中有多少元素,即为该层有多少个数。
cnt--
;cnt
等于 0 时,则该层数据访问完毕。C++
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> ans;
if(root == nullptr) return ans;
queue<TreeNode*> q;
q.push(root);
while(!q.empty()){
int cnt = q.size();
vector<int> temp;
while(cnt-- > 0){
TreeNode* cur = q.front();
q.pop();
temp.push_back(cur->val);
if(cur->left != nullptr) q.push(cur->left);
if(cur->right != nullptr) q.push(cur->right);
}
ans.push_back(temp);
}
return ans;
}
};
Java
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> ans = new ArrayList<List<Integer>>();
if(root == null) return ans;
Queue<TreeNode> q = new LinkedList<>();
q.add(root);
while(!q.isEmpty()){
int cnt = q.size();
ArrayList<Integer> temp = new ArrayList<>();
while(cnt-- > 0){
TreeNode cur = q.poll();
temp.add(cur.val);
if(cur.left != null) q.add(cur.left);
if(cur.right != null) q.add(cur.right);
}
ans.add(temp);
}
return ans;
}
}
n
为树上所有节点的个数,每个点进队出队各一次,故渐进时间复杂度为 O ( n ) O(n) O(n)。n
个,故渐进空间复杂度为 O ( n ) O(n) O(n)。题目来源:力扣。
放弃一件事很容易,每天能坚持一件事一定很酷,一起每日一题吧!
关注我LeetCode主页 / CSDN—力扣专栏,每日更新!