题目:输入一颗二元树,从上往下按层打印树的每个结点,同一层中按照从左往右的顺序打印。
例如输入
思路:我们从树的根结点开始分析。自然先应该打印根结点8,同时为了下次能够打印8的两个子结点,我们应该在遍历到8时把子结点6和10保存到一个数据容器中。现在数据容器中就有两个元素6 和10了。按照从左往右的要求,我们先取出6访问。打印6的同时要把6的两个子结点5和7放入数据容器中,此时数据容器中有三个元素10、5和7。接下来我们应该从数据容器中取出结点10访问了。注意10比5和7先放入容器,此时又比5和7先取出,就是我们通常说的先入先出。因此不难看出这个数据容器的类型应该是个队列。
既然已经确定数据容器是一个队列,现在的问题变成怎么实现队列了。实际上我们无需自己动手实现一个,因为STL已经为我们实现了一个很好的deque(两端都可以进出的队列),我们只需要拿过来用就可以了
- #include <deque>
- #include <iostream>
- using namespace std;
- struct BTreeNode // a node in the binary tree
- { int m_nValue; // value of node
- BTreeNode *m_pLeft; // left child of node
- BTreeNode *m_pRight; // right child of node
- };
- void PrintFromTopToBottom(BTreeNode *pTreeRoot)
- {
- if(pTreeRoot==NULL)
- return;
- deque<BTreeNode *> queue;
- queue.push_back(pTreeRoot);
- while(queue.size())
- {
- BTreeNode *now=queue.front();
- cout<<now->m_nValue<<" ";
- queue.pop_front();
- if(now->m_pLeft!=NULL)
- queue.push_back(now->m_pLeft);
- if(now->m_pRight!=NULL)
- queue.push_back(now->m_pRight);
- }
- }
- void main()
- {
- BTreeNode a[7];
- a[0].m_nValue=8;
- a[1].m_nValue=6;
- a[2].m_nValue=10;
- a[3].m_nValue=5;
- a[4].m_nValue=7;
- a[5].m_nValue=9;
- a[6].m_nValue=11;
- a[0].m_pLeft=&a[1];
- a[0].m_pRight=&a[2];
- a[1].m_pLeft=&a[3];
- a[1].m_pRight=&a[4];
- a[2].m_pLeft=&a[5];
- a[2].m_pRight=&a[6];
- a[3].m_pLeft=NULL;
- a[3].m_pRight=NULL;
- a[4].m_pLeft=NULL;
- a[4].m_pRight=NULL;
- a[5].m_pLeft=NULL;
- a[5].m_pRight=NULL;
- a[6].m_pLeft=NULL;
- a[6].m_pRight=NULL;
- PrintFromTopToBottom(&a[0]);
- }