题解 | #从上往下打印二叉树#

从上往下打印二叉树

https://www.nowcoder.com/practice/7fe2212963db4790b57431d9ed259701

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
#include <cstdio>
#include <queue>
#include <vector>
class Solution {
public:
    vector<int> PrintFromTopToBottom(TreeNode* root) {
		vector<int> res;
		if (root == nullptr)
			return res;

		queue<TreeNode*> queue;
		queue.push(root);

		int nextLevel = 0;
		int toBePrinted = 1;

		while (!queue.empty()) {
			TreeNode *pTreeNode = queue.front();
			res.push_back(pTreeNode->val);
			
			queue.pop();
			toBePrinted--;

			if (pTreeNode->left != nullptr) {
				queue.push(pTreeNode->left);
				nextLevel++;
			}
			if (pTreeNode->right != nullptr) {
				queue.push(pTreeNode->right);
				nextLevel++;
			}

			if (toBePrinted == 0) {
				printf("\n");
				toBePrinted = nextLevel;
				nextLevel = 0;
			}
		}
		return res;
    }
};

全部评论

相关推荐

05-19 19:57
蚌埠学院 Python
2237:Gpa70不算高,建议只写排名,个人技能不在多而在精,缩到8条以内。项目留一个含金量高的,减少间距弄到一页,硕士简历也就一页,本科不要写很多
实习,投递多份简历没人回...
点赞 评论 收藏
分享
星辰再现:裁员给校招生腾地方
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务