队列的最大值(C++)

题目:

请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。

若队列为空,pop_front 和 max_value 需要返回 -1

思路:

使用队列实现,最大队列最前端始终保持当前最大值及其index,如果新加入的值大于当前队列中的值,则清空当前小于当前值得所有值,保存当前值到队列后端,保存当前最大值,如果小于则保存到最大队列的后边(最大值被弹出后,当前值为最大值),如果队列中当前最大值被弹出,根据相应的index判断是否为最大队列的头值

代码实现:

class MaxQueue {
public:
    MaxQueue():cur_idx(0) { // 新加入的值在队列中的index

    }
    
    int max_value() {
        if(max_data.empty())
            return -1;
        return max_data.front().data_val;
    }
    
    void push_back(int value) {
        while(!max_data.empty() && value >= max_data.back().data_val)
            max_data.pop_back(); // 弹出最大队列中小于当前的值
        temp temp_data{value, cur_idx};
        data.push_back(temp_data);
        max_data.push_back(temp_data);
        cur_idx++;
    }
    
    int pop_front() {
        if(max_data.empty())
            return -1;
        int result = data.front().data_val;
        if(data.front().index == max_data.front().index)
            max_data.pop_front();  // 根据队列中弹出的值的index判断是否需要弹出最大队列中的值
        data.pop_front();
        return result;
    }

    struct temp
    {
        int data_val;
        int index;
    };
    int cur_idx;
    deque<temp> data;
    deque<temp> max_data;


};
发布了63 篇原创文章 · 获赞 4 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/Jerry_Leo_liu/article/details/105338245