Binary Tree Zigzag Level Order Traversal -- LeetCode

Given a binary tree, return the zigzag level order traversal of its nodes' values. (ie, from left to right, then right to left for the next level and alternate between).

For example:
Given binary tree {3,9,20,#,#,15,7},

    3
   / \
  9  20
    /  \
   15   7

 

return its zigzag level order traversal as:

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

思路：方法一是每一层都按照从左到右的顺序存入结果中，最后将所有奇数层(root为第0层)反转。时间复杂度O(n)

class Solution {
public:
    void help(vector<vector<int> >& res, TreeNode* root, int depth)
    {
        if (!root) return;
        if (res.size() < depth + 1)
            res.push_back(vector<int>(1, root->val));
        else res[depth].push_back(root->val);
        help(res, root->left, depth + 1);
        help(res, root->right, depth + 1);
    }
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        vector<vector<int> > res;
        help(res, root, 0);
        for (int i = 1, n = res.size(); i < n; i += 2)
            reverse(res[i].begin(), res[i].end());
        return res;
    }
};

 

第二种方法：使用一个队列。依次遍历每一层。使用一个bool类型变量来记录当前层是需要从左向右还是从右向左。  每一层在队列中时都是从左向右的遍历顺序，如果当前层就是需要从左向右，则直接存入结果；否则，计算出该节点应该在的下标，并将它存入。

用队列遍历每一层的方法是，在进入每一层的最开始，当前队列中的节点全是该层的节点，此时队列中的元素数就是该层应该有的元素数。若该层需要从右向左遍历时，则每个元素在结果中的下标应该是size - i - 1。

该方法的时间复杂度同样是O(n).

class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        vector<vector<int> > res;
        if (!root) return res;
        queue<TreeNode *> q;
        q.push(root);
        bool LoR = true;
        while (!q.empty())
        {
            int size = q.size();
            vector<int> cand(size);
            for (int i = 0; i < size; i++)
            {
                int index = LoR ? i : size - i - 1;
                TreeNode* cur = q.front();
                q.pop();
                cand[index] = cur->val;
                if (cur->left) q.push(cur->left);
                if (cur->right) q.push(cur->right);
            }
            res.push_back(cand);
            LoR = !LoR;
        }
        return res;
    }
};