層序遍歷二叉樹
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]
]
之前也有見過這個題,按照層序打印出二叉樹,如果只是打印,就比較簡單。
思路如下:
先創建一個隊列,首先是將根節點插入到隊列中,得到隊列首元素,打印出該元素的值,再將該首元素的左節點與右節點依次插入到隊列中,將該首元素刪除出隊列。再取隊列首元素,打印值,插入其非空孩子節點即可。
但是今天碰見的題是需要將該二叉樹的值按照層序的順序插入到二維vector中。
顯然這個題和上面的類似,還是需要藉助到隊列來完成。
還需要在每次循環每層的時候創建一個一維數組來存儲這一層的元素的值,在這次遍歷結束後,將該一維數組尾插到最終的二維數組後。
首先還是應該將根節點插入到隊列中,取出隊列首元素,將該元素的值放在臨時的一維數組中,再將該元素的非空左孩子有孩子節點插入到隊列中,刪除首元素。再循環,直到隊列爲空,結束循環。得到最後的結果。
需要注意的是:在判斷是否該到下一層時,我們需要定義一個臨時的隊列來暫時存儲下一層的節點,爲什麼?因爲有了這個臨時的隊列,我們可以只判斷隊列q是否爲空,若爲空,則表示該層遍歷結束,該遍歷下一層了。 這時我們再將臨時的隊列與q交換再繼續插入判斷即可。
具體實現的代碼如下:
/**
* Definition for binary tree
* 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 *head) {
vector<vector <int> > ret;
queue<TreeNode*> q;
queue<TreeNode*> tmp;
if(head == NULL)
return ret;
q.push(head);
while(!q.empty())
{
vector<int> v;
while(!q.empty())
{
TreeNode* front = q.front();
v.push_back(front->val);
if(front->left)
tmp.push(front->left);
if(front->right)
tmp.push(front->right);
q.pop();
}
swap(tmp,q);
ret.push_back(v);
}
return ret;
}
};