剑指offer——34之字打印二叉树

题目描述

请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

 

题解：

　　与上道题没区别，就是在存入数据时，对于奇数行的数据，先反转一下，再存入即可

　　

class Solution {
public:
    vector<vector<int> > Print(TreeNode* pRoot) {
        vector<vector<int>>res;
        BFS(pRoot, res);
        return res;
    }
    void BFS(TreeNode *root, vector<vector<int>>&res)
    {
        if (root == nullptr)return;
        queue<TreeNode*>q;
        q.push(root);
        bool fromLeft = true;
        while (!q.empty())
        {
            queue<TreeNode*>temp;
            vector<int>v;
            while (!q.empty())
            {
                TreeNode* p = q.front();
                q.pop();
                v.push_back(p->val);
                if (p->left != nullptr)temp.push(p->left);
                if (p->right != nullptr)temp.push(p->right);
            }
            if(fromLeft)
                res.push_back(v);
            else
            {
                reverse(v.begin(), v.end());
                res.push_back(v);
            }
            fromLeft = !fromLeft;
            q = temp;
        }
    }
};