23、从上到下打印二叉树 III
请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3 / \ 9 20 / \ 15 7
返回其层次遍历结果:
[ [3], [20,9], [15,7] ]
提示:
节点总数 <= 1000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof
思路:
参考:https://blog.csdn.net/m1015422754/article/details/112292891
思路一模一样,不过需要用一个变量记录一下是哪个方向,如果从左到右,我们则无需变化,如果从右到左,我们只需要把记录的数组翻转一下再放进答案即可。
代码:
/**
* 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) {
vector<vector<int>> res;
queue<TreeNode*> q;
if(!root) return res;
q.push(root);
int dt = 1; //1代表左到右,0代表右到左
while(q.size())
{
vector<int> level;
//层次遍历一整层
int len = q.size();
while(len --)
{
auto t = q.front();
q.pop();
level.push_back(t->val);
if(t->left) q.push(t->left);
if(t->right) q.push(t->right);
}
if(!dt) reverse(level.begin(), level.end()); //从右到左,翻转数组
dt ^= 1; //变换方向
res.push_back(level); //进入答案
}
return res;
}
};