代码随想录栈与队列3

最新推荐文章于 2024-09-06 10:38:06 发布

乌拉！哒哒哒哒哒

最新推荐文章于 2024-09-06 10:38:06 发布

阅读量150

点赞数 1

分类专栏：代码随想录打卡记录文章标签：算法数据结构 leetcode

本文链接：https://blog.csdn.net/sunke123456789/article/details/136464416

版权

代码随想录打卡记录专栏收录该内容

30 篇文章 0 订阅

订阅专栏

文章介绍了两个C++编程问题解决方案：使用deque实现的滑动窗口最大值查找，以及使用unordered_map和priority_queue找到输入整数数组中的前k个高频元素。

摘要由CSDN通过智能技术生成

239 滑动窗口最大值

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        vector<int> result;
        deque<int> window; // 存储元素的下标
        int n = nums.size();
        
        for (int i = 0; i < n; ++i) {
            // 如果窗口的首部元素已经不在滑动窗口内，则移除它
            if (!window.empty() && window.front() == i - k)
                window.pop_front();
            
            // 保持窗口中元素的递减顺序
            while (!window.empty() && nums[i] >= nums[window.back()])
                window.pop_back();
            
            // 将当前元素下标加入窗口
            window.push_back(i);
            
            // 如果当前窗口已经形成，将窗口最大值加入结果
            if (i >= k - 1)
                result.push_back(nums[window.front()]);
        }
        
        return result;
    }
};

347前k个高频元素

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> freqMap;
        for (int num : nums) {
            freqMap[num]++;
        }
        
        // 定义优先队列，按照频率降序排列
        auto cmp = [](const pair<int, int>& a, const pair<int, int>& b) {
            return a.second < b.second;
        };
        priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(cmp)> pq(cmp);
        
        // 将元素和频率加入优先队列
        for (auto& entry : freqMap) {
            pq.push(entry);
        }
        
        // 取出前 k 个频率最高的元素
        vector<int> result;
        for (int i = 0; i < k; ++i) {
            result.push_back(pq.top().first);
            pq.pop();
        }
        
        return result;

    }
};