二叉树按层次遍历,并且打印每行结点个数

和层次遍历类似,利用queue性质可以实现。问题关键在于每行开始结束的结点在哪里,只要找到每行开始或结束的地方就可以打印换行,实现按层遍历。
思路:用两个指针,last和nlast,last记录正在打印的行的最后一个位置,
nlast记录下一行的最后一个位置(如果当前行是最后一行,则nlast不存在)。现在就是怎么找last和nlast了,如果当前打印的位置来到了last,则说明当前行已经打印完毕,应该换行,此时让last = nlast,开始打印下一行。
nlast则在各结点入队的时候更新,下一行最后一个入队结点作为nlast,依次进行。
在这里插入图片描述

这里vectorv是记录每行结点个数,思路就是从这次换行到下次换行的时候多少个结点出队了,就是该行结点个数,每个统计完一行count应该从新赋值为0

void print_By_Level(btnode* root, vector<int>&v) {
	btnode *last, *nlast;
	queue<btnode*>m_que;
	last = nlast = root;
	m_que.push(last);
	int count = 0;
	while (!m_que.empty()){	
		btnode *temp = m_que.front();
		m_que.pop();
		count++;   //每次出队列说明该行个数+1
		if (temp->lchild!=NULL) {
			m_que.push(temp->lchild);
			nlast = temp->lchild;//nlast一直保持在该行的最后一个位置
		}
		if (temp->rchild!=NULL) {
			m_que.push(temp->rchild);
			nlast = temp->rchild;
		}
		if (temp != last) {
			cout << temp->data <<" ";
		}
		else {                  //当前位置来到last 换行 开始打印下一行 下一行的nlast变为该行的last
			v.push_back(count);//每次来到last 说明该行打印完毕 count赋值0 继续计算下一行结点个数
			count = 0;
			cout << temp->data << endl;
			last = nlast;
		}
	}
}

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_42673507/article/details/85521379
今日推荐