之字形打印二叉树
题目
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推
思路
这题和从上到下打印二叉树类似https://www.cnblogs.com/tianzeng/p/10186431.html;需要两个栈;打印某一层结点时,把下一层结点保存到相应的栈里
- 如果当前层是奇数层,先保存左子结点再保存右子结点到第一个栈里
- 如果当前层是偶数层,先保存右子结点再保存左子结点到第二个栈里
输出顺序
1
3 2
4 5 6 7
15 14 13 12 12 10 9 8
为什么用两个栈?
如果只有一个栈,在打印根结点的时候,先后把2和3保存到栈里,在打印栈顶元素3的时候会把3的子节点(如果有的话)保存到栈里,这样以来,栈顶元素就不是2,无法打印元素2
/** * 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); bool is_left = false; while (!q.empty()) { int len = q.size(); vector<int> tmp; for (int i = 0; i < len; ++i) { TreeNode *node = q.front(); q.pop(); tmp.push_back(node->val); //也可奇数行从左向右入队,偶数行从右向左入队 if (node->left != nullptr) { q.push(node->left); } if (node->right != nullptr) { q.push(node->right); } } if (is_left) { res.push_back(vector<int>(tmp.rbegin(), tmp.rend())); } else { res.push_back(tmp); } is_left = !is_left; } return res; } };