从上到下打印二叉树
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回:
[3,9,20,15,7]
提示:
- 节点总数 <= 1000
解法:使用一个队列(先进先出,从队尾进入,队首出来),来存储结点。每次将队首元素的左右结点入队列,然后将队首元素出队列。直到队列为空。
之前学数据结构时,学了对应的理论,现在将当初的思想写出来别有一番滋味。
/**
* 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<int> levelOrder(TreeNode* root) {
if(root == nullptr)
return vector(0,0); // 确定返回值
deque<TreeNode*> deque1;
vector<int> result;
deque1.push_back(root);
while(!deque1.empty()){
if(deque1.front()->left != nullptr)
deque1.push_back(deque1.front()->left);
if(deque1.front()->right != nullptr)
deque1.push_back(deque1.front()->right);
result.push_back(deque1.front()->val);
deque1.pop_front();
}
return result;
}
};