牛客網劍指offer-從上往下列印二叉樹
阿新 • • 發佈:2019-01-07
題目描述
從上往下打印出二叉樹的每個節點,同層節點從左至右列印。/* struct TreeNode { int val; struct TreeNode *left; struct TreeNode *right; TreeNode(int x) : val(x), left(NULL), right(NULL) { } };*/ class Solution { public: //BFS vector<int> PrintFromTopToBottom(TreeNode* root) { vector<int> ans; if (root == nullptr) return ans; queue<TreeNode*> q; q.push(root); while (!q.empty()) { TreeNode *p = q.front(); ans.push_back(p->val); if (p->left != nullptr) q.push(p->left); if (p->right != nullptr) q.push(p->right); q.pop(); } return ans; } };