今天做了一道“南湖旁的树”,考察到二叉树层次遍历(宽度优先),一般用队列可实现,但要求按行输出。在书上的程序都是依次便利而没有按行输出,修改了一番,每次队列非空就都pop出来再push新的。因为二叉树嘛,最多也就pop两次,而且题目要求每行从小到大,还需要排序,也简单,就两个比较。
void bfs(){
int m,n,tmp;
int i=1;
q.push(1);
while(!q.empty())
{
cout<<i++<<":";
m=q.front();q.pop();
if(!q.empty()){
n=q.front();q.pop();
if(n>m){tmp=n;n=m;m=tmp;}
cout<<n<<" ";
if(v[n].left) q.push(v[n].left);
if(v[n].right) q.push(v[n].right);
}
cout<<m;
if(v[m].left) q.push(v[m].left);
if(v[m].right) q.push(v[m].right);
cout<<"\n";
}
注:程序用的是数组模拟二叉树而没用指针,因为根据原题数组会简单点,但思想是一样的。框架中队列的应用简直精髓!