从上往下打印二叉树

题目描述

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

因为对STL的队列不熟悉，这里用vector来实现队列的功能：先进先出

下面有用队列实现的，更加简洁

/* 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) { //存储要打印的每个节点的内容 vector<int> printVec; //存储要打印的每个节点的指针 vector<TreeNode*> nodeVec; if(root == nullptr) return printVec; //将根节点加入 nodeVec.push_back(root); //若当前节点有左子节点或者右子节点，则将其加入（先左后右），同时将当前节点的内容加入printVec //然后指向nodeVec中的下一个节点 for(int i = 0; i < nodeVec.size(); i++){ if(nodeVec[i]->left != nullptr) nodeVec.push_back(nodeVec[i]->left); if(nodeVec[i]->right != nullptr) nodeVec.push_back(nodeVec[i]->right); printVec.push_back(nodeVec[i]->val); } return printVec; } };

用队列queue实现，更简洁：

/* 链接：https://www.nowcoder.com/questionTerminal/7fe2212963db4790b57431d9ed259701 来源：牛客网 */ class Solution { public: vector<int> PrintFromTopToBottom(TreeNode* root) { vector<int> res; if(root==NULL) return res; queue<TreeNode*> q; q.push(root); while(!q.empty()){ res.push_back(q.front()->val); if(q.front()->left!=NULL) q.push(q.front()->left); if(q.front()->right!=NULL) q.push(q.front()->right); q.pop(); } return res; } };