【C++算法】队列相关经典算法题

1. N叉树的层序遍历

首先我们遇到这个题目，没有任何思路，我们就可以来模拟一下层序的流程，首先我们肯定是访问根节点1，访问之后呢就是访问下一层的最左节点3，此时第一层的节点1已经访问过了就可以不要了，然后访问第二层的中间节点2，最后访问最右边的节点4，然后就访问第三层，此时第二层最做节点就不要了，此时我们会发现此时就满足队列的先进先出的特点，知道用什么数据结构了我们就直接上思路：

直接上代码：

class Solution {
public:
    vector<vector<int>> levelOrder(Node* root) {
        vector<vector<int>> ret; //记录最终结果
        queue<Node*> q;  // 层序遍历需要的队列
        if(root == nullptr)
            return ret;
        q.push(root); // 队头元素入队
        while(!q.empty())
        {
            vector<int> tmp; //统计本层的节点
            int size = q.size(); //求出本层元素的个数
            while(size--)
            {
                Node* n = q.front();
                q.pop();
                tmp.push_back(n->val);
                for(auto child : n->children) // 让孩子入队列
                {
                    if(child != nullptr)
                        q.push(child);
                }
            }
            ret.push_back(tmp);
        }
        return ret;
    }
};

2. 二叉树的锯齿形层序遍历

这个题目上给我们说了层序遍历，那么我们还是要用到队列，我们会发现，这道题和我们上道题目是一样的都是层序遍历，唯一的区别就是偶数层的遍历方式是我们之前的逆序，所以这道题目也很简单，直接上思路：

直接来写代码：

class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        vector<vector<int>> ret;
        queue<TreeNode*> q;
        int flag = 1;//增加一个标记位,让偶数行的信息逆序即可
        if(root == nullptr) return ret;
        q.push(root);
        while(!q.empty())
        {
            vector<int> tmp;
            int size = q.size(); 
            while(size--)
            {
                TreeNode* front = q.front();
                q.pop();
                tmp.push_back(front->val);
                if(front->left != nullptr)
                    q.push(front->left);
                if(front->right != nullptr)
                    q.push(front->right);
            }
            if(flag == 1)
                ret.push_back(tmp);
            else
            {
                reverse(tmp.begin(),tmp.end());
                ret.push_back(tmp);
            }  
            flag *= -1;
        }
        return ret;
    }
};

3. 在每个树行中找最大值

我们这道题和之前的题目思路是一样的，只不过这个题目我们不需要统计每层的节点，只需要统计最大的哪一个节点即可，所以在层序遍历过程中，在执⾏让下⼀层节点⼊队的时候，我们可以在循环中统计出当前层结点的最大值的，直接上思路：

直接来写代码：

class Solution {
public:
    vector<int> largestValues(TreeNode* root) {
        vector<int> ret;
        if(root == nullptr) return ret;
        queue<TreeNode*> q;
        q.push(root);
        while(q.size()) // 如果队列不为空
        {
            int maxnum = INT_MIN;
            int size = q.size();
            while(size--)
            {
                TreeNode* t = q.front();
                q.pop();
                maxnum = max(maxnum, t->val); // 记录每层的最大值
                if(t->left) q.push(t->left);
                if(t->right) q.push(t->right);
            }
            ret.push_back(maxnum);
        }
        return ret;
    }
};

4.二叉树的最大宽度

题目解析：

这道题相较于上面三道题就有点难度了，这个题目男难就难在处理空节点，既然统计每一层的最大宽度，我们优先想到的就是利用层序遍历，把当前层的结点全部存在队列⾥面，利用队列的常度来计算每一层的宽度，统计出最大的宽度。但是，由于空节点也是需要计算在内的。因此，我们可以选择将空节点也存在队列⾥⾯。

⭐解法一：

这个思路是我们正常会想到的思路，但是极端境况下，最左边⼀条长链，最右边⼀条长链，我们需要存几亿个空节点，会超过最⼤内存限制。

⭐解法二：利用数组存储二叉树的方式，给节点编号

依旧是利⽤层序遍历，但是这⼀次队列⾥⾯不单单存结点信息，并且还存储当前结点如果在数组中存储所对应的下标（在我们学习数据结构堆的时候，计算左右孩子的方式）。

这样我们计算每⼀层宽度的时候，⽆需考虑空节点，只需将当层结点的左右结点的下标相减再加 1即可，我们直接上思路： 

魔鬼细节：

极端境况下，最左边⼀条长链，最右边⼀条长链，我们需要存几亿个空节点，会超过最⼤内存限制，那么此时我们的编号是2的1500次方，那么肯定会越界，此时无论什么数据类型我们都存不下，但是这个题最后我们是会进行减法的，但是我们数据的存储是⼀个环形的结构，并且题⽬说明，数据的范围在 int 这个类型的最大值的范围之内，因此不会超出⼀圈； 因此，如果是求差值的话，我们⽆需考虑溢出的情况，由于c++溢出使用int会报错，因此我们使用unsigned int来存储这个数据。直接上代码：

class Solution {
public:
    int widthOfBinaryTree(TreeNode* root) {
        queue<pair<TreeNode*, unsigned int>> q;
        q.push({root,1}); // 根节点和编号入队
        unsigned int ret = 0; // 最大宽度
        while(q.size())
        {
            // 先更新这⼀层的宽度
            auto& [x1, y1] = q.front();
            auto& [x2, y2] = q.back();
            ret = max(ret, y2 - y1 + 1);

            int size = q.size();
            // 让下⼀层进队
            while(size--)
            {
                auto [a,b] = q.front(); 
                q.pop();
                if(a->left) q.push({a->left, 2*b});
                if(a->right) q.push({a->right,2*b+1});
            }
        }
        return ret;
    }
};

其实这道题我们也可以用数组来实现，但是这时候有人说了此时我们会经常进行删除节点，而删除节点都是头删，而数组的头删效率较低，你咋还使用我们的数组呢？确实是这样，但是我们进行删除节点不一定要头删，我们可以把本层的节点用tmp统计出来，然后拷贝给q即可。

class Solution {
public:
    int widthOfBinaryTree(TreeNode* root) {
        vector<pair<TreeNode*,unsigned int>> q; // 用数组模拟队列 

        q.push_back(make_pair(root,1));
        unsigned int ret = 0;

        while(!q.empty())
        {
            // 先更新这⼀层的宽度
            auto& [x1, y1] = q[0];
            auto& [x2, y2] = q.back();
            ret = max(ret, y2 - y1 + 1);
            // 让下⼀层进队
            vector<pair<TreeNode*, unsigned int>> tmp; // 让下⼀层进⼊这个队列
            for(auto [x, y] : q)
            {
                if(x->left) tmp.push_back({x->left, y * 2});
                if(x->right) tmp.push_back({x->right, y * 2 + 1});
            }
            q = tmp; // 避免了头删效率低的问题
        }
        return ret;
    }
};