题解 | 按之字形顺序打印二叉树
按之字形顺序打印二叉树
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) {} * }; */ class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param pRoot TreeNode类 * @return int整型vector<vector<>> */ vector<vector<int> > Print(TreeNode* pRoot) { // write code here vector<vector<int>> res; if (pRoot == nullptr) return res; deque<TreeNode*> q; q.push_back(pRoot); int level = 1; while (!q.empty()) { vector<int> cur; for (int i = q.size(); i > 0; i--) { if (level % 2 == 1) { TreeNode* node = q.front(); q.pop_front(); cur.push_back(node->val); if (node->left != nullptr) { q.push_back(node->left); } if (node->right != nullptr) { q.push_back(node->right); } } else { TreeNode* node = q.back(); q.pop_back(); cur.push_back(node->val); if (node->right != nullptr) { q.push_front(node->right); } if (node->left != nullptr) { q.push_front(node->left); } } } res.push_back(cur); level++; } return res; } };