题目链接
剑指 Offer 32 - III. 从上到下打印二叉树 III
实现思路
- 这道题和实现二叉树层序(题目链接:102. 二叉树的层序遍历)基本一样,只是多了奇偶层数判断。
- 每一次循环的开始,队列的大小就是当前层数的所有节点个数(按照从左到右的顺序);
依次把当前节点的左右孩子入队:if (temp->left) { q.push(temp->left); } if (temp->right) { q.push(temp->right); }
- 使用一个
flag
标识当前层数是奇数还是偶数。if (flag) { reverse(vec.begin(), vec.end()); flag = false; } else { flag = true; }
实现代码(C++)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
if (root == NULL) {
return vector<vector<int>>();
}
queue<TreeNode*> q;
q.push(root);
vector<vector<int>> ans;
bool flag = false;
while (!q.empty()) {
int size = q.size();
vector<int> vec(size);
for (int i = 0; i < size; i++) {
TreeNode *temp = q.front();
q.pop();
vec[i] = temp->val;
if (temp->left) {
q.push(temp->left);
}
if (temp->right) {
q.push(temp->right);
}
}
if (flag) {
reverse(vec.begin(), vec.end());
flag = false;
} else {
flag = true;
}
ans.push_back(vec);
}
return ans;
}
};