利用队列的FIFO特性来实现层序遍历。
初始化:创建队列q,root入队。
只要队列不为空:存队首元素,把队首元素pop出来,打印队首元素。判断它是否有左孩子,如果有,则左孩子入队;判断是否有右孩子,如果有,则右孩子入队(先判断左孩子再右孩子是因为这里一层中遍历假定从左到右)。
有点像利用栈的前序遍历,但是栈是LIFO,因此假如要进行前序遍历,则入栈的时候是先判断并入右孩子”go” command,再判断并入左孩子”go” command,最后入当前节点”print” command。
而且,只要队列不为空,每一次取出来队首元素一定会先打印,这就是层序遍历;而在栈不为空的时候,每一次取出栈顶元素要先判断当前命令是否是”print”,只有print的时候才打印,否则是”go”,就是继续向下深度遍历,之后依次弹出这一趟遍历的元素的command,再进行判断。
struct TreeNode{
int val;
TreeNode* left;
TreeNode* right;
};
void levelTraversal(TreeNode* root){
if(root == NULL)//处理root为空的情况
return;
queue<TreeNode*> q;//注意这里是TreeNode*类型不是int
q.push(root);//队列初始化完成
while(!q.empty()){//当队列不为空
TreeNode* node = q.front();
q.pop();
cout<<node->val;//打印当前节点
if(node->left)//左孩子不为空则入队
q.push(node->left);
if(node->right)//右孩子不为空则入队
q.push(node->right);
}
}