有一棵二叉树,请设计一个算法,按照层次打印这棵二叉树。
给定二叉树的根结点root,请返回打印结果,结果按照每一层一个数组进行储存,所有数组的顺序按照层数从上往下,且每一层的数组内元素按照从左往右排列。保证结点数小于等于500。
具体要求如图所示:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class TreePrinter {
public:
vector<vector<int> > printTree(TreeNode* root) {
// write code here
TreeNode *last = root;//last表示正在打印的当前行的最右节点
TreeNode *nlast = root;//表示下一行的最右节点
TreeNode *tmp = NULL;
queue<TreeNode*> q;
vector<vector<int> > ans;
vector<int> ret;
if(root)
q.push(root);
while(!q.empty()){
tmp = q.front();
q.pop();
ret.push_back(tmp->val);
if(tmp->left){
q.push(tmp->left);
nlast = tmp->left;
}
if(tmp->right){
q.push(tmp->right);
nlast = tmp->right;
}
//当当前节点等于当前行的最右节点时,更新last
//把当前行的所有数据放到二维数组ans中,并且清空存储每行数据的数组ret
if(tmp == last){
last = nlast;
ans.push_back(ret);
ret.clear();
}
}
return ans;
}
};