103. Binary Tree Zigzag Level Order Traversal

image.png

解法一:使用队列,读的时候用典型的BFS即可。
但是保存答案的时候,奇数不用翻转,偶数翻转。
注意reverse的用法。void reverse()

/**
 * 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>> zigzagLevelOrder(TreeNode* root) {
        vector<vector<int>> result;
        if(root == NULL) return result;
        vector<int> cur;
        queue<vector<TreeNode*>> q;
        cur.push_back(root->val);
        vector<TreeNode*> vcur;
        vcur.push_back(root);
        q.push(vcur);
        int flag = 1;
        result.push_back(cur);
        while(!q.empty()){
            vector<TreeNode*> myFront = q.front();
            q.pop();
            vector<int> list;
            vector<TreeNode*> plist;
            flag *= -1;
            for(int i = 0; i < myFront.size(); i++){
                if(myFront[i]->left){
                    list.push_back(myFront[i]->left->val);
                    plist.push_back(myFront[i]->left);
                }
                if(myFront[i]->right){
                    list.push_back(myFront[i]->right->val);
                    plist.push_back(myFront[i]->right);
                }
            }
            if(flag == -1){
                reverse(list.begin(), list.end());
            }
            if(plist.size()){
                q.push(plist);
                result.push_back(list);
            }

        } 
        return result;
    }
   
};

解法二:用queue

/**
 * 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>> zigzagLevelOrder(TreeNode* root) {
        queue<TreeNode*> q;
        vector<vector<int>> result;
        if(root == NULL) return result;
        q.push(root);
        bool flag = true;
        int index;
        while(!q.empty()){
            int size = q.size();
            vector<int> list(size);
            for(int i = 0; i < size; i++){
                TreeNode* node = q.front();
                q.pop();
                index = flag ? i : size - 1 - i;
                list[index] = node->val;
                if(node->left) q.push(node->left);
                if(node->right) q.push(node->right);
            }
            flag = !flag;
            result.push_back(list);
        }
        return result;
    }
};
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 很多人会有一个困惑,就是书读了那么多,为什么感觉没有效果?我想,原因无非有几点:一是没有记笔记,二是没有读透书,三...
    Hikari阅读 494评论 0 5
  • 这个个案是个spa馆老板,也是一个芳疗爱好者,应她的要求,她店里的几个员工全程在旁边观摩,让我开始有点紧张,有些分...
    木木尘非阅读 359评论 0 0
  • 坚持自己的本心真的很难(是对我个人而言,其他人我不知道啦),有些书本传递给我这样一种观点:坚持自己,相信自己所做...
    团子君阅读 266评论 0 1