本文已参与「新人创作礼」活动,一起开启掘金创作之路。
LeetCode-剑指 Offer 32 - III. 从上到下打印二叉树 III
这道题在 问题 上更近一步,不仅要分层打印,还要相邻层的打印顺序是相反的,其实只需要加一个标志flag,每遍历一层取反,就能标识相邻层。
/**
* 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 == nullptr) return {};
vector<vector<int>> res;
queue<TreeNode*> q;
q.push(root);
TreeNode* tempNode = root;
TreeNode* rightNode = tempNode;
vector<int> ans;
bool flag = true;
while(!q.empty()) {
TreeNode* t = q.front();
q.pop();
ans.push_back(t->val);
if(t->left != nullptr) {
q.push(t->left);
tempNode = t->left;
}
if(t->right != nullptr) {
q.push(t->right);
tempNode = t->right;
}
if(rightNode == t) {
if(!flag) reverse(ans.begin(), ans.end());
flag = !flag;
res.push_back(ans);
ans.clear();
rightNode = tempNode;
}
}
return res;
}
};
\