您的位置:首页 > 其它

打印二叉树中第m层第k个节点(递归+非递归)

2016-06-03 16:10 246 查看
前面的文章中已经讨论过“ 层序遍历二叉树指定的某层”的问题,在这个基础上,对代码稍作修改,即可完成“打印二叉树中第m层第k个节点”,具体实现如下。

说明:根节点位于0层,每层中的第一个节点的下标为0;所以如果要打印二叉树中第m层第k个节点(需要传入的参数分别为:(m-1) 和 (k-1))。

#include <iostream>
using namespace std;
#include <vector>
#include <deque>

typedef struct node
{
char data;
struct node *lchild;
struct node *rchild;
}BiNode, *BiTree;

// 先序建立二叉树 (输入时,按先序次序输入二叉树中结点的值,以 # 字符表示空树)
BiTree createBiTree()
{
BiTree T;

char c;
scanf("%c", &c);

if (c == '#')
T = NULL;
else
{
T = new BiNode;  // 或 T = (BiTree)malloc(sizeof(BiNode));
T->data = c;

T->lchild = createBiTree();
T->rchild = createBiTree();
}

return T;

}

// 递归,输出level层下标为k的节点。
// 参数level代表几层(不是第几层),k代表该层中要访问的节点对应的下标;(如第3层,第2个节点 应传入参数(2, 1))
int k = 1; // k=1 表示要访问level层的第2个节点
int num = 0; // 初始化num,当访问到对应的level层时,num用于统计该层节点对应的下标,当num==k时,即找到目标节点。
void PrintNodeAtLevel(BiTree T , int level)
{
if(T == NULL || level < 0) return;

if(level == 0)
{
if (num == k)
printf("%c\n", T->data);

num++;

}
PrintNodeAtLevel(T->lchild , level - 1);
PrintNodeAtLevel(T->rchild , level - 1);
}

// 非递归(一),此处借助vector来实现,输出level层下标为k的节点。
// 参数level代表几层(不是第几层),k代表该层中要访问的节点对应的下标;(如第3层,第2个节点 应传入参数(2, 1))
void printNodeAtLevel2(BiTree T, int level, int k) // 参数level代表几层(不是第几层)
{
if (T == NULL) return;

vector<BiTree> vec;
vec.push_back(T);

int curLevel = 0; // 当前位于几层;(注意是几层,不是第几层;例如,根节点位于0层,即第1层)

int cur = 0;
int last = 1;
while (cur < vec.size() && curLevel < level)
{
last = (int)vec.size();

while (cur < last)
{
if (vec[cur]->lchild)
vec.push_back(vec[cur]->lchild);

if (vec[cur]->rchild)
vec.push_back(vec[cur]->rchild);

cur++;
}

curLevel++; // 当前层数加1
}

// 跳出上面的while循环时,curLevel == level;然后直接打印出当前层的下标位于[cur + k]的那个节点即可。
printf("%c\n", vec[cur + k]->data);

}

// 非递归(二),此处借助queue来实现,输出level层下标为k的节点。
// 参数level代表几层(不是第几层),k代表该层中要访问的节点对应的下标;(如第3层,第2个节点 应传入参数(2, 1))
void printNodeAtLevel3(BiTree T, int level, int k)
{
if(T == NULL) return;

deque<BiTree> q;
q.push_back(T);

int curLevelNum; // 当前层的节点数

int curLevel = 0; // 当前位于几层

while (q.size() && curLevel < level)
{
curLevelNum = (int)q.size(); // 此处要做类型转换,因为size()返回值为size_type类型;

while(curLevelNum-- > 0) // 一直访问到当前层的最后一个节点
{
BiTree tmp = q.front();

q.pop_front();

if(tmp->lchild)
q.push_back(tmp->lchild);
if(tmp->rchild)
q.push_back(tmp->rchild);

}

curLevel++; // 当前层数加1
}

// 跳出上面的while循环时,curLevel == level;然后直接打印出当前层的下标是k的那个节点即可。
printf("%c\n", q[k]->data);
}

int main(int argc, const char * argv[]) {

BiTree T = createBiTree(); // 建立

PrintNodeAtLevel(T, 2); // 递归输出(第3层第2个节点)(注意:访问第几个节点,实在上面int k = 1处指定的)

//    printNodeAtLevel2(T, 2, 1); // 非递归
//
//    printNodeAtLevel3(T, 2, 1); // 非递归

return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息