给你二叉树的根节点?root
?,返回其节点值的?锯齿形层序遍历?。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
vector<vector<int>> v;
if(root == nullptr) return v;
queue<TreeNode*> q;
q.push(root);
bool flag = 0;//多加一个flag
while(!q.empty())
{
int size = q.size();
vector<int> vec;
while(size-- > 0)
{
TreeNode* node = q.front();
q.pop();
vec.push_back(node->val);
if(node->left) q.push(node->left);
if(node->right) q.push(node->right);
}
if(flag)
reverse(vec.begin(), vec.end());//flag为1时翻转此次vec
flag = !flag;
v.push_back(vec);
}
return v;
ps:刚开始,还在想使用stack,但是发现stack只能压到栈顶,还得需要一个vec辅助进行,难搞,还是算法库好哇!reverse!!!