leetcode剑指offer32-III 广度优先算法

题目：https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof/
思路：广度优先算法
利用栈（先进后出）

1. 定义两个栈，一个用于存储当前节点，一个用于存储下一层的节点
2. 判断根节点是否为零，为零就将返回空；
3. 将根节点入栈1，将标识符设置为1，（表示对栈1 内的节点进行操作）
4. 栈1、栈2不为空 执行循环体
5. 如果标识符为1，计算栈1节点的个数，将每个节点从栈中抛出，每抛出一个节点，将结点的值入一维数组，将节点的左子树入栈2，将节点的右子树入栈2，（左右子树入栈的顺序有要求），遍历完栈1中的节点后，将一维数组入二维数组，并将一维数组清空，将标识符设置为2；
6. 如果标识符为2，计算栈2点的个数，将每个节点从栈中抛出，每抛出一个节点，将结点的值入一维数组，将节点的左子树入栈1，将节点的右子树入栈1，（左右子树入栈的顺序有要求），遍历完栈2中的节点后，将一维数组入二维数组，并将一维数组清空，将标识符设置为1；
7. 直到所有的节点都遍历（栈1和栈2都为空);

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {


        vector<vector<int>> vec;
        stack<TreeNode*>stk1,stk2;


        int flag = 1;
        if (root == NULL)
            return vec;
        stk1.push(root);
        vector<int>A;
        while (!stk1.empty()|| !stk2.empty())
        {
            
            if (flag == 1)
            {
                int len = stk1.size();
                for (int i = 0; i < len; i++)
                {
                    TreeNode*node = stk1.top();
                    stk1.pop();
                    A.push_back(node->val);
                    if (node->left)
                        stk2.push(node->left);
                    if (node->right)
                        stk2.push(node->right);


                }
                vec.push_back(A);
                A.clear();
                flag = 2;
            }
            else
            {
                int len = stk2.size();
                for (int i = 0; i < len; i++)
                {
                    TreeNode*node = stk2.top();
                    stk2.pop();
                    A.push_back(node->val);
                    if (node->right)stk1.push(node->right);
                    if (node->left) stk1.push(node->left);
                }
                vec.push_back(A);
                A.clear();
                flag = 1;
            }
        }
        return vec;
    }
};