题目
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路
举例说明:
1、8入队列 deque: 8
2、8的左右孩子依次入队列 deque: 8 6 10
3、存8,8出队列 deque: 6 10
4、6的左右孩子依次入队列 deque: 6 10 5 7
5、存6,6出队列 deque: 10 5 7
6、10的左右孩子依次入队列 deque: 10 5 7 9 11
7、存10,10出队列 deque: 5 7 9 11
.......
#include <iostream>
#include <deque>
#include <vector>
using namespace std;
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
class Solution {
public:
//层次遍历,运用双端队列
vector<int> PrintFromTopToBottom(TreeNode* head) {
vector<int> result;
if (head == nullptr)
return result;
deque<TreeNode *> d;
d.push_back(head);
while (d.size())
{
if (d.front()->left != nullptr)
d.push_back(d.front()->left);
if (d.front()->right != nullptr)
d.push_back(d.front()->right);
result.push_back(d.front()->val);
//cout << d.front()->val << endl;
d.pop_front();
}
return result;
}
};
int main()
{
Solution s;
TreeNode *head = new TreeNode(8);
TreeNode *head1 = new TreeNode(6);
head->left = head1;
TreeNode *head2 = new TreeNode(10);
head->right = head2;
TreeNode *head3 = new TreeNode(5);
head2->left = head3;
vector<int> result = s.PrintFromTopToBottom(head);
for (auto i : result)
cout << i << endl;
system("pause");
return 0;
}
deque用法:
deque容器为一个给定类型的元素进行线性处理,像向量一样,它能够快速地随机访问任一个元素,并且能够高效地插入和删除容器的尾部元素。但它又与vector不同,deque支持高效插入和删除容器的头部元素,因此也叫做双端队列。deque类常用的函数如下。
(1) 构造函数
deque():创建一个空deque
deque(int nSize):创建一个deque,元素个数为nSize
deque(int nSize,const T& t):创建一个deque,元素个数为nSize,且值均为t
deque(const deque &):复制构造函数
(2) 增加函数
void push_front(const T& x):双端队列头部增加一个元素X
void push_back(const T& x):双端队列尾部增加一个元素x
iterator insert(iterator it,const T& x):双端队列中某一元素前增加一个元素x
void insert(iterator it,int n,const T& x):双端队列中某一元素前增加n个相同的元素x
void insert(iterator it,const_iterator first,const_iteratorlast):双端队列中某一元素前插入另一个相同类型向量的[forst,last)间的数据
(3) 删除函数
Iterator erase(iterator it):删除双端队列中的某一个元素
Iterator erase(iterator first,iterator last):删除双端队列中[first,last)中的元素
void pop_front():删除双端队列中最前一个元素
void pop_back():删除双端队列中最后一个元素
void clear():清空双端队列中最后一个元素
(4) 遍历函数
reference at(int pos):返回pos位置元素的引用
reference front():返回手元素的引用
reference back():返回尾元素的引用
iterator begin():返回向量头指针,指向第一个元素
iterator end():返回指向向量中最后一个元素下一个元素的指针(不包含在向量中)
reverse_iterator rbegin():反向迭代器,指向最后一个元素
reverse_iterator rend():反向迭代器,指向第一个元素的前一个元素
(5) 判断函数
bool empty() const:向量是否为空,若true,则向量中无元素
(6) 大小函数
Int size() const:返回向量中元素的个数
int max_size() const:返回最大可允许的双端对了元素数量值
(7) 其他函数
void swap(deque&):交换两个同类型向量的数据
void assign(int n,const T& x):向量中第n个元素的值设置为x
参考:https://blog.csdn.net/longshengguoji/article/details/8519812