题目:
今天这题,难的并不是算法思想,而是对树的操作熟不熟悉。
题目的意思很简单,做法也很简单,就是层序遍历加上奇数列的时候逆序输出。那么不论是用什么数据结构做辅助,只要满足要求就可以。
由于思路很简单,直接上C++代码了:
但是树的创建和操作我还不是很熟,所以这次先不贴出
class Solution {
public:
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
vector<vector<int>> ans;
if (!root) {
return ans;
}
queue<TreeNode*> nodeQueue;
nodeQueue.push(root);
bool isOrderLeft = true;//锯齿队列标记
while (!nodeQueue.empty()) {//层序遍历
deque<int> levelList;//记录当前列
int size = nodeQueue.size();
for (int i = 0; i < size; ++i) {
auto node = nodeQueue.front();
nodeQueue.pop();//队列的弹出是从前面弹出
if (isOrderLeft) {
levelList.push_back(node->val);
} else {
levelList.push_front(node->val);
}
if (node->left) {//为下一次遍历做准备
nodeQueue.push(node->left);
}
if (node->right) {
nodeQueue.push(node->right);
}
}
ans.emplace_back(vector<int>{levelList.begin(), levelList.end()});//这种初始化方法学到了
isOrderLeft = !isOrderLeft;//取反
}
return ans;
}
};