给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回锯齿形层次遍历如下:
[
[3],
[20,9],
[15,7]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
#include <algorithm>
class Solution {
public:
struct mem{
TreeNode* root;
int layer;
};
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
if(root==NULL)
{
return {};
}
mem mRoot;
mRoot.root=root;
mRoot.layer=0;
queue<mem> q;
q.push(mRoot);
int vSize=0;
vector<vector<int>> v;
while(!q.empty())
{
mem curRoot=q.front();
q.pop();
if(curRoot.root->left!=NULL)
{
mem tmpRoot;
tmpRoot.root=curRoot.root->left;
tmpRoot.layer=curRoot.layer+1;
q.push(tmpRoot);
}
if(curRoot.root->right!=NULL)
{
mem tmpRoot;
tmpRoot.root=curRoot.root->right;
tmpRoot.layer=curRoot.layer+1;
q.push(tmpRoot);
}
if(vSize<curRoot.layer+1)
{
vSize++;
v.resize(vSize);
v[curRoot.layer].push_back(curRoot.root->val);
}
else
{
v[curRoot.layer].push_back(curRoot.root->val);
}
}
for(int i=0; i<v.size(); i++)
{
if(i%2==1)
{
reverse(v[i].begin(),v[i].end());
}
}
return v;
}
};