本系列文章已全部上傳至我的github,地址:ZeeCoder‘s Github
歡迎大家關注我的新浪微博,我的新浪微博
歡迎轉載,轉載請注明出處
來源: https://leetcode.com/problems/binary-tree-level-order-traversal/
Given a binary tree, return the level order traversal of its nodes’ values. (ie, from left to right, level by level).
For example:
Given binary tree [3,9,20,null,null,15,7],3
/ \
9 20
/ \
15 7
return its level order traversal as:[
[3],
[9,20],
[15,7]
]
本題大意:給定1個2叉樹,按層的順序輸出各層元素
解題思路:每次將該層元素壓入1個queue,然后順次處理這個queue里的每一個元素,并把它們的左右節點都壓入另外一個queue,1直這樣處理,直到queue為空。
/**
* 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>> ret;
if(root==NULL) return ret;
queue<TreeNode*> que;//用來寄存每層的節點
que.push(root);//先壓入根節點
while(!que.empty())
{
vector<int> temp;
queue<TreeNode*> tmpQue;//寄存下1層queue
while(!que.empty())
{
TreeNode* tempNode = que.front();//順次取出queue里的元素進行處理
que.pop();
temp.push_back(tempNode->val);//將值壓入vector
if(tempNode->left!=NULL) tmpQue.push(tempNode->left);//處理左節點
if(tempNode->right!=NULL) tmpQue.push(tempNode->right);//處理右節點
}
ret.push_back(temp);//將每層的結果壓入返回vector中
que=tmpQue;.
}
return ret;
}
};
下一篇 序列化二叉樹