Leetcode刷题笔记103. 二叉树的锯齿形层序遍历

时间：2020年12月22日
知识点：二叉树、广度优先
题目链接：https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal/

题目
给定一个二叉树，返回其节点值的锯齿形层序遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。

例如：
给定二叉树 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回锯齿形层序遍历如下：

[
  [3],
  [20,9],
  [15,7]
]

解法：

1. 这是一道普通的层序遍历 升级版 在原来只要一层层遍历的时候 锯齿形输出
2. 考虑到他的大体框架 还是不变的 就是一个队列往里面放节点
3. 变化的是 放入答案的顺序 这里用双向队列 能从前面放 也能从后面放
4. 锯齿形 说明方向要有变化 用flag标记 这些节点从头放 还是添加在尾部
5. 最后用deque转成vector 变成答案

代码

#include <stdio.h>
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        queue<TreeNode*> q;
        vector<vector<int>> ans;
        if(root == nullptr)
            return ans;
        q.push(root);
        int flag = 0;
        while(!q.empty()){
            deque<int> d;
            int num = q.size();
            while(num--){
                TreeNode *tmp = q.front();
                q.pop();
                if(flag == 0)
                    d.push_back(tmp->val);
                else if(flag == 1)
                    d.push_front(tmp->val);
                if(tmp->left)
                    q.push(tmp->left);
                if(tmp->right)
                    q.push(tmp->right);
            }
            flag = !flag;
            ans.push_back(vector<int>(d.begin(),d.end()));
        }
        return ans;
    }
};
int main()
{
    Solution s;
    TreeNode root(3);
    TreeNode node1(9),node2(20),node3(15),node4(7);
    root.left = &node1,root.right = &node2;
    node2.left = &node3;node2.right = &node4;
    vector<vector<int>> ans = s.zigzagLevelOrder(&root);
    for(vector<int> v : ans){
        for(int x : v)
            cout<<x<<" ";
        cout<<endl;
    }
    return 0;
}

今天也是爱zz的一天哦！