[劍指offer]從上到下打印二叉樹
題目描述
從上到下打印出二叉樹的每個節點,同一層的節點按照從左到右的順序打印。
例如:
給定二叉樹: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回:
[3,9,20,15,7]
提示:
節點總數 <= 1000
通過次數22,621提交次數35,155
解題思路
- 用一個隊列進行層次遍歷
- 先將二叉樹的頭節點入隊,然後出隊,然後判斷是否有左子樹,有的話入隊,是否有右子樹,有的話入隊。然後出隊,對出隊節點進行訪問,如此反覆,直到隊列爲空。
實現代碼
class Solution {
public:
vector<int> levelOrder(TreeNode* root) {
vector<int>res;
queue<TreeNode*> que;
que.push(root);
while(!que.empty()){
TreeNode* temp = que.front();//獲得隊列頭節點
que.pop();
if(temp){//節點不爲null時
res.push_back(temp->val);//節點值保存
que.push(temp->left);//左子樹入隊
que.push(temp->right);//右子樹入隊
}
}
return res;
}
};