二叉树层次遍历就是按每层从左到右(一般是从左到右,若想从右到左也很简单)的次序遍历结点。下面是一个简单的例子:
这棵二叉树层次遍历的结果是 1,2,3,4,5。实现层次遍历一般是用队列,思路还是比较简单:
(1) 首先把根结点入队。
(2) 若队列不为空,则将队头结点出队,访问该结点,并判断出队结点的左右孩子,若左孩子非空,则左孩子入队,若右孩子非空,则右孩子也入队。
(3) 重复2)直到队列为空。
那么对于上面那棵树,其层次遍历的过程我用下图来表示:
下面给出代码:
#include <iostream>
#include <queue>
using namespace std;
struct TreeLinkNode {
int val;
TreeLinkNode *left, *right, *next;
TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
};
void levelTraversal(TreeLinkNode *root) {
if(NULL==root)
return;
queue<TreeLinkNode*> myQueue;
myQueue.push(root);
while(!myQueue.empty()){
TreeLinkNode *node=myQueue.front();
myQueue.pop();
cout<<node->val<<endl;
if(NULL!=node->left)
myQueue.push(node->left);
if(NULL!=node->right)
myQueue.push(node->right);
}
}
int main()
{
TreeLinkNode *root=new TreeLinkNode(1);
TreeLinkNode *left=new TreeLinkNode(2);
TreeLinkNode *right=new TreeLinkNode(3);
TreeLinkNode *leftRight=new TreeLinkNode(4);
TreeLinkNode *rightRight=new TreeLinkNode(5);
root->left=left;
root->right=right;
left->right=leftRight;
right->left=rightRight;
levelTraversal(root);
system("pause");
return 0;
}
下面说一个稍微麻烦一丁点儿的问题,就是上面遍历结点时,其实并不知道访问的结点是哪一层的。
如果我在访问一个结点的时候,想知道这个结点是哪一层的,怎么办?
我的解决方法是:在一层结束的时候,把一个特殊标记入队,用于标记一层已经结束了,那么当这个特殊标记出队时,我就知道即将遍历到开始新一层的结点了。
我这里所用的特殊标记是
NULL,下面继续先用图来阐释一下,有图有真相。。还是刚才那棵树的例子(画图还是挺累的,嘿嘿>_<):
呼呼呼,终于画完图了(>_<)。。。啥也不说了,上代码。。。
#include <iostream>
#include <queue>
using namespace std;
struct TreeLinkNode {
int val;
TreeLinkNode *left, *right, *next;
TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
};
void levelTraversal(TreeLinkNode *root) {
if(NULL==root)
return;
queue<TreeLinkNode*> myQueue;
myQueue.push(root);
myQueue.push(NULL);
int level=1;
while(!myQueue.empty()){
TreeLinkNode *node=myQueue.front();
myQueue.pop();
if(NULL==node){
if(myQueue.empty())
break;
level++;
myQueue.push(NULL);
continue;
}
cout<<"第"<<level<<"层结点:"<<node->val<<endl;
if(NULL!=node->left)
myQueue.push(node->left);
if(NULL!=node->right)
myQueue.push(node->right);
}
}
int main()
{
TreeLinkNode *root=new TreeLinkNode(1);
TreeLinkNode *left=new TreeLinkNode(2);
TreeLinkNode *right=new TreeLinkNode(3);
TreeLinkNode *leftRight=new TreeLinkNode(4);
TreeLinkNode *rightRight=new TreeLinkNode(5);
root->left=left;
root->right=right;
left->right=leftRight;
right->left=rightRight;
levelTraversal(root);
system("pause");
return 0;
}
这样就达到效果了:
如果有更好的方法,欢迎交流~