打印二叉树中第K层的第M个节点,非递归算法
1、二叉树定义:
typedef struct BTreeNodeElement_t_ {
void *data;
} BTreeNodeElement_t;
typedef struct BTreeNode_t_ {
BTreeNodeElement_t *m_pElemt;
struct BTreeNode_t_ *m_pLeft;
struct BTreeNode_t_ *m_pRight;
} BTreeNode_t;
2、求二叉树中第K层的第M个节点
(1)非递归算法
借助队列实现
首先将给定根节点pRoot入队:
第一步:如果队列未空,获取当前队列的长度,即当前层的节点总数;
第二步:记录当前遍历的层数,判断是否超出指定层数,如果超出则退出;如果小于指定层数,则对当前层的所有左右节点入队操作;如果等于指定 层数,则进行第三步;
第三步:获取当前队列中节点总数,如果当前节点总数小于指定节点数,则退出;如果节点总数大于指定节点数,则进行第四步;
第四步:遍历当前层节点,如果节点数等于指定节点数,则放回此节点。
第三步:循环结束后,如果没有符合条件的节点就返回NULL。
BTreeNode_t * GetKthLevelMthNode( BTreeNode_t *pRoot, int KthLevel, int MthNode){
if( pRoot == NULL || KthLevel <= 0 || MthNode <= 0 )
return NULL;
queue <BTreeNode_t *> que;
que.push( pRoot );//首先将根节点入队
int level = 0; //当前层计数器
int cntNode = 0; //当前层节点数计数器
int curLevelNodesTotal = 0;//当前层节点总数
while( !que.empty() ){
++level;
if( level > KthLevel)//如果层数已大于指定层数,则退出
break;
cntNode = 0; //当前层节点数计数器归0
curLevelNodesTotal = que.size();//当前层的节点总数
while( cntNode < curLevelNodesTotal ){
++cntNode;//记录当前层的节点数
pRoot = que.front();
que.pop();
if( level == KthLevel && cntNode == MthNode ){ //看当前节点的层数和在当前层中的节点次序是否符合要求
break;
}
//将当前层节点的左右结点均入队,即将下一层节点入队
if( pRoot->m_pLeft )
que.push( pRoot->m_pLeft);
if( pRoot->m_pRight)
que.push( pRoot->m_Right);
}
if( level == KthLevel && cntNode == MthNode ){ //看当前节点的层数和在当前层中的节点次序是否符合要求
break;
}
}
while( !que.empty()){//清空栈
que.pop();
}
if( level == KthLevel && cntNode == MthNode ){ //看当前节点的层数和在当前层中的节点次序是否符合要求
return pRoot;
}
return NULL;
}