题目地址:牛客网
描述
给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)
数据范围:0n≤1500,树上每个节点的val满足 |val|<=100
要求:空间复杂度:O(n),时间复杂度:O(n)
例如:
给定的二叉树是{1,2,3,#,#,4,5}
该二叉树之字形层序遍历的结果是
[
[1],
[3,2],
[4,5]
]
示例1
输入:
{1,2,3,#,#,4,5}
返回值:
[[1],[3,2],[4,5]]
说明:
如题面解释,第一层是根节点,从左到右打印结果,第二层从右到左,第三层从左到右。
示例2
输入:
{8,6,10,5,7,9,11}
返回值:
[[8],[10,6],[5,7,9,11]]
示例3
输入:
{1,2,3,4,5}
返回值:
[[1],[3,2],[4,5]]
解题思路:
一开始又是队列又是栈地写了一段,觉得没有错但是答案一直不对)
然后发现了可以直接用reverse反转vector。
于是直接层次遍历就好啦。
代码:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
class Solution {
public:
vector<vector<int> > Print(TreeNode* pRoot) {
TreeNode* p=pRoot;
vector<vector<int> >v;
if(!p)return v;
queue<TreeNode*> q;
q.push(p);
int i=0;
while(q.size())
{
int j=q.size();
vector<int> vv;
while(j--){
p=q.front();
q.pop();
vv.push_back(p->val);
if(p->left)q.push(p->left);
if(p->right)q.push(p->right);
}
i++;
if(i%2==0){
reverse(vv.begin(),vv.end());
}
v.push_back(vv);
}
return v;
}
};