题目链接:牛客网
二叉树的层次遍历是队列queue这个数据结构的一个经典应用场景之一,注意队列里的元素类型是节点指针(不知道并且有兴趣的可以想想为什么~)
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
vector<int> PrintFromTopToBottom(TreeNode *root) {
queue<TreeNode*> q;
vector<int> v;
if (root == NULL)
return v;
q.push(root);
while (!q.empty()) {
TreeNode* now = q.front();
q.pop();
v.push_back(now->val);
if (now->left != NULL)
q.push(now->left);
if (now->right != NULL)
q.push(now->right);
}
return v;
}
};
上面那个“queue里的元素为何指针比较好”的问题的一个可能的回答(不一定对,仅个人理解)——因为需要拿到节点的值,需要将节点的孩子节点加入队列以待后续访问,而为什么不存一个对象呢?浪费空间呗,指针的size就几个字节(32位机器上是4字节,64位机器是8字节),而一个对象一般都比指针的size要大,用指针划算。