LeetCode 103——二叉树的锯齿形层次遍历

1. 题目

2. 解答

定义两个栈 s_l_r、s_r_l 分别负责从左到右和从右到左遍历某一层的节点，用标志变量 flag 来控制具体情况，根节点所在层 flag=1 表示从左到右遍历，每隔一层改变一次遍历方向。

用栈 s_l_r 从左到右遍历当前层节点时，按照先左子节点再右子节点的顺序将这一层节点的子节点依次放入栈 s_r_l 中。

用栈 s_r_l 从右到左遍历当前层节点时，按照先右子节点再左子节点的顺序将这一层节点的子节点依次放入栈 s_l_r 中。

/**
 * 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;
        vector<int> temp;
        stack<TreeNode *> s_l_r;
        stack<TreeNode *> s_r_l;
        if (root)   s_l_r.push(root);
        int flag = 1; 
        // 根节点层从左往右遍历，然后每隔一层改变遍历方向
        
        while (!s_l_r.empty() || !s_r_l.empty())
        {
            if (flag)
            {
                while (!s_l_r.empty())
                {
                    TreeNode * cur = s_l_r.top();
                    s_l_r.pop();
                    temp.push_back(cur->val);
                    if (cur->left) s_r_l.push(cur->left);
                    if (cur->right) s_r_l.push(cur->right);
                } 
            }
            else
            {
                while (!s_r_l.empty())
                {
                    TreeNode * cur = s_r_l.top();
                    s_r_l.pop();
                    temp.push_back(cur->val);
                    if (cur->right) s_l_r.push(cur->right);
                    if (cur->left) s_l_r.push(cur->left);
                } 
            }
            flag = 1 - flag;
            result.push_back(temp);
            temp.clear();
        }
        return result;
    }
};

获取更多精彩，请关注「seniusen」!