从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
class Solution {
public:
/*
思路:改编于面试题32,利用队列。
先打印根节点,并把根节点左右子节点入队,打印队首元素并出队,将其左右孩子入队...循环如此一直到队列为空。
此题要分层打印,则添加一个一维vector数组levelPrint存储每层的节点,再增设一个指针rightSon赋值为队列qTree尾元素,指向每层的最右节点
当当前指针tNode与rightSon指向同一节点,表示该层遍历完毕,打印该层并重新设置levelPrint和rightSon节点
*/
vector<vector<int> > Print(TreeNode* root) {
//将vector作为返回值类型
vector<vector<int> > printTree;
//若为空树,直接返回空的数组
if(root==nullptr)
return printTree;
//因为要分层打印,所以定义辅助指针指向每层最右边的节点,表示这是这一层的结尾
TreeNode *rightSon=root;
//储存每层的节点,最后加入到二维vector printTree中
vector<int> levelPrint;
//建立辅助队列
queue<TreeNode*> qTree;
TreeNode *tNode=root;
//根节点入队
qTree.push(tNode);
//若队列不为空且树指针不指向空,执行循环:
while(!qTree.empty()&&tNode!=nullptr)
{
//将队列首元素放入打印数组中
levelPrint.push_back(tNode->val);
//若队手元素左右子树不为空,压入队列
if(tNode->left)
qTree.push(tNode->left);
if(tNode->right)
qTree.push(tNode->right);
//若此节点等于当前层最右节点,则将这一层结果输出并将最右节点重新赋值为队列尾结点
if(rightSon==tNode)
{
printTree.push_back(levelPrint);
levelPrint.clear();
rightSon=qTree.back();
}
//队首元素出队
qTree.pop();
//tNode赋值为新的队首元素
tNode=qTree.front();
}
return printTree;
}
};