#采用层次遍历方法设计求二叉树的第k层的结点个数
#这个二叉树仍然是二叉链存储结构,层次遍历我认为是在二叉链存储结构的基础上进行改进得来的
#层次遍历的算法用到了队列,队列的每一个结点都有两个属性:该结点所在的层级、访问的存储结构结点的地址
#层级的设置:一旦结点访问了其左孩子结点或者右孩子结点其所在的层级就要+1
#队列用于存储留待判断的结点
#总体计数的思路:如果结点的层级数与要求的相等,计数器就+1,如果层次数小于要求的层数,就把这个结点的左孩子和右孩子结点进队(并在这个过程中把层级+1),留待之后判断
#如果层级数大于要求的层级数,返回计数器,留待下一个被判断的结点使用计数器
#采用层次遍历方法设计求二叉树的第k层的结点个数
#这个二叉树仍然是二叉链存储结构,层次遍历我认为是在二叉链存储结构的基础上进行改进得来的
#层次遍历的算法用到了队列,队列的每一个结点都有两个属性:该结点所在的层级、访问的存储结构结点的地址
#层级的设置:一旦结点访问了其左孩子结点或者右孩子结点其所在的层级就要+1
#队列用于存储留待判断的结点
#总体计数的思路:如果结点的层级数与要求的相等,计数器就+1,如果层次数小于要求的层数,就把这个结点的左孩子和右孩子结点进队(并在这个过程中把层级+1),留待之后判断
#如果层级数大于要求的层级数,返回计数器,留待下一个被判断的结点使用计数器
from collections import deque
class QNode:
def __init__(self,l,p):
self.lev=l
self.node=p
def KCount1(bt,k):
cnt=0
qu=deque()
qu.append(QNode(l,bt.b))
while len(qu)>0:
p=qu.popleft()#出一个结点并用p指针指向它,用来判断
if p.lev>k:
return cnt
if p.lev==k:
cnt+=1
else:#添加左右孩子结点进队
if p.node.lchild!=None:
qu.append(QNode(p.lev+1,p.node.lchild))
if p.node.rchild!=None:
qu.append(QNode(p.lev+1,p.node.rchild))