层序遍历利用BFS搜索算法,利用queue,写了c++和Java两个版的。
c++中队列queue< T >的pop()方法返回void类型而非T类型,要先用front()获取
(c++)代码如下:
vector<int> PrintFromTopToBottom(TreeNode* root) {
queue<TreeNode *> q;
vector<int> rst;
if(root==NULL){
return rst;
}
q.push(root);
while(!q.empty()){
int size=q.size();
for(int i=0;i<size;i++){
TreeNode *cur=q.front();
q.pop();
rst.push_back(cur->val);
if(cur->left!=NULL){
q.push(cur->left);
}
if(cur->right!=NULL){
q.push(cur->right);
}
}
}
return rst;
}
java中队列Queue的方法简述:
add 增加一个元索,如果队列已满,则抛出一个IIIegaISlabEepeplian异常
remove 移除并返回队列头部的元素,如果队列为空,则抛出一个NoSuchElementException异常
element 返回队列头部的元素, 如果队列为空,则抛出一个NoSuchElementException异常
offer 添加一个元素并返回true,如果队列已满,则返回false
poll 移除并返问队列头部的元素, 如果队列为空,则返回null
peek 返回队列头部的元素, 如果队列为空,则返回null
put 添加一个元素,如果队列满,则阻塞
take 移除并返回队列头部的元素,如果队列为空,则阻塞
(Java)代码如下:
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> rsts=new ArrayList<>();
if(root==null)
return rsts;
Queue<TreeNode> q=new LinkedList<TreeNode>();
q.offer(root);
while(!q.isEmpty()){
ArrayList<Integer> rst=new ArrayList<>();
int size=q.size();
for(int i=0;i<size;i++){
TreeNode head=q.poll();
rst.add(head.val);
if(head.left!=null)
q.offer(head.left);
if(head.right!=null)
q.offer(head.right);
}
rsts.add(rst);
}
return rsts;
}