题解 | #按之字形顺序打印二叉树#
按之字形顺序打印二叉树
https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0
/** * struct TreeNode { * int val; * struct TreeNode *left; * struct TreeNode *right; * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * }; */ #include <algorithm> #include <queue> class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param pRoot TreeNode类 * @return int整型vector<vector<>> */ vector<vector<int> > Print(TreeNode* pRoot) { // write code here vector< vector<int> > ret; if (pRoot == nullptr) { return ret; } queue<TreeNode *> qe; qe.push(pRoot); bool flag = true; while (!qe.empty()) { vector<int> arr; int size = qe.size(); for (int i = 0; i < size; ++i) { TreeNode * ptr = qe.front(); arr.push_back(ptr->val); if (ptr->left != nullptr) { qe.push(ptr->left); } if (ptr->right != nullptr) { qe.push(ptr->right); } qe.pop(); } if (!flag) { reverse(arr.begin(), arr.end()); } ret.push_back(arr); flag = !flag; } return ret; } };
在层次遍历的基础上, 每隔一行将输出的元素进行反转即可实现效果