[leetcode]剑指 Offer 32 - III. 从上到下打印二叉树 III BFS二叉树

Posted 2022-03-08 PushyTao

题目链接

二叉树简单遍历

将访问的序列存入$vector$，对于每一层交替从左向右从右向左我们可以设置一个变量flag来决定方向
用$pre$表示上一层的节点数量，用$cnt$求得上本层的节点数量，在遍历的时候遇到一个点就将$pre-1$，如果说$pre$为0了，那么就说明上一层的节点已经访问完毕，然后将在该过程中访问得到$val$的$vector$集合存入答案，根据$flag$的值进行判断是否需要逆置

Code:

/**
 * 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>> levelOrder(TreeNode *root) 
        vector <vector<int>> ret;
        if (root == NULL) return ret;
        queue < TreeNode * > que;
        vector<int> t;
        int pre = 1;
        que.push(root);
        int cnt = 0;
        int flag = 1;
        while (que.size()) 
            TreeNode *node = que.front();
            que.pop();
            pre--;
            t.push_back(node->val);
            if (node->right != NULL) 
                que.push(node->right);
                cnt++;
            
            if (node->left != NULL) 
                que.push(node->left);
                cnt++;
            
            if (pre == 0) 
                pre = cnt;
                cnt = 0;
                if (flag == -1)
                    ret.push_back(t);
                else 
                    reverse(t.begin(), t.end());
                    ret.push_back(t);
                
                t.clear();
                flag *= -1;
            
        
        return ret;
    
;