百度笔试题------层次遍历二叉树分析

给定以下二叉树:

struct node_t

{

    node_t *left, *right;

    int value;

};

要求编写函数 node_t* foo(node_t *node, unsigned int m, unsigned int k);

输出以 node 为根的二叉树第 m 层的第 k 个节点值.

(level, k 均从 0 开始计数)

注意：

.此树不是完全二叉树；

.所谓的第K个节点，是本层中从左到右的第K个节点 

这是百度的一道笔试题，很明显，从题目中可以看出，考察的是层次遍历二叉树的知识点。

层次遍历二叉树，需要利用队列，其实和图的BFS算法差不多。只不过图的广度优先遍历的一个节点有多个子节点，而对于一个二叉树而言，它的一个节点最多只有2个子节点，比BFS还要简单。

这道题需要求第m层的第k个子节点，那么如何判断遍历的层数？这是一个最为关键的问题。

一个队列要有头指针和尾指针，当一层的节点入队列后，就用last变量记录，这样就可以知道自己遍历到了哪一层。

至于求每一层遍历的结点个数，非常简单，只要到新的一层就从0开始数即可。

PS：我的代码与题目要求稍有不同，比如我的leve和num是从1开始计数，函数也略有改动。

不过，总的思想是一致的。

#include<iostream> #define MAX 100 using namespace std; typedef struct node_t { int value; struct node_t *left,*right; }*BiTree,TreeNode; BiTree queue[MAX]; int top=-1,rear=-1; int level=1,num=0; bool QueueIsEmpty() { if(top==rear) { return true; } else { return false; } } void EnQueue(BiTree value) { queue[++rear]=value; } BiTree DeQueue() { return queue[++top]; } void CreateTree(BiTree &root) { char ch; cin>>ch; root=new TreeNode(); if(ch=='/') { root=NULL; return; } root->value=ch; CreateTree(root->left); CreateTree(root->right); } BiTree foo(BiTree node, unsigned int m, unsigned int k) { BiTree p=node; int last; if(p!=NULL) { EnQueue(p); while(!QueueIsEmpty()) { last=rear; while(top<last) { BiTree q=DeQueue(); num++; if(level==m&&num==k) { cout<<q->value<<endl; break; } if(q->left!=NULL) { EnQueue(q->left); } if(q->right!=NULL) { EnQueue(q->right); } } level++; num=0; } } cout<<"没有这个节点"<<endl; return p; } int main() { BiTree root; CreateTree(root); cout<<"建立树完成"<<endl; foo(root,4,3); system("pause"); return 0; }