描述
给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)
数据范围:0 \le n \le 15000≤n≤1500,树上每个节点的val满足 |val| <= 1500∣val∣<=1500
要求:空间复杂度:O(n)O(n),时间复杂度:O(n)O(n)
例如:
给定的二叉树是{1,2,3,#,#,4,5}
该二叉树之字形层序遍历的结果是
[
[1],
[3,2],
[4,5]
]
思路:先层次遍历,遇到偶数行就翻转。
class Solution {
public:
vector<vector<int> > Print(TreeNode* pRoot) {
// write code here
vector<vector<int>> res;
if(pRoot==nullptr)return res;
queue<TreeNode*> q;
q.push(pRoot);
int sentence=0;
TreeNode* cur;
while(!q.empty())
{
vector<int> row;
int n=q.size();
sentence++;
for(int i=0;i<n;i++)
{
cur=q.front();
q.pop();
row.push_back(cur->val);
//向队列里插入子节点
if(cur->left!=nullptr)q.push(cur->left);
if(cur->right!=nullptr)q.push(cur->right);
}
if(sentence%2==0)reverse(row.begin(),row.end());
res.push_back(row);
}
return res;
}
};
思路:层次遍历,用队列辅助,先把头结点插入队列,统计队列大小,作为循环次数,再循环清空当前队列到vector中,再把当前元素集的子节点插入到空队列中,依次循环,Z字顺序只需要在偶数行用reverse函数交换队列顺序。