请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
思路:层次遍历,但是偶数层的需要反着添加!
class Solution {
public:
void reverse(vector<int>&vec)
{
for(int i=0;i<vec.size()/2;i++)
{
swap(vec[i],vec[vec.size()-1-i]);
}
}
vector<vector<int> > Print(TreeNode* pRoot) {
vector<vector<int>>res;
vector<int>vec;
queue<TreeNode *>que;
if(pRoot)
{
que.push(pRoot);
int t=0;
while(!que.empty())
{
int n=que.size();
for(int i=0;i<n;i++)
{
TreeNode *node=que.front();
que.pop();
vec.push_back(node->val);
if(node->left)
que.push(node->left);
if(node->right)
que.push(node->right);
}
if(t%2==0)
res.push_back(vec);
else
{
reverse(vec);
res.push_back(vec);
}
vec.clear();
t++;
}
}
return res;
}
};