LeetCode 239滑动窗口最大值 347前k个高频元素 | 代码随想录25期训练营day12

LeetCode 239 滑动窗口最大值 2023.11.6

• 题目链接
• 代码随想录讲解[链接]
  
• 题意：输入一个整数数组，然后用一个大小为k的窗口依次从数组开始滑向末尾，输出每个窗口内的最大值保存到一个数组中，最后输出该数组
• 方法：对窗口内的数据结构及接口自定义操作，例如窗口滑动时，删除窗口前一个元素需判断其是否为窗口的最大值，添加窗口尾元素时，先将窗口内比该值小的都删除，最后放入该元素(排序)，窗口滑动时记录最大值到返回的数组中

class Solution {
public:
    //对删除操作进行定义，删除需判断队列是否为空且要删除的元素是否为该窗口的最大值
    void pop(deque<int> &dq, int val)
    {
        if(!dq.empty() && val == dq.front())
            dq.pop_front();
    }
    //对添加操作进行定义，对窗口内元素进行排序，当窗口内元素小于要加入元素时删除，最后加入该元素
    void push(deque<int> &dq, int val)
    {
        while (!dq.empty() && dq.back() < val)
            dq.pop_back();
        dq.push_back(val);
    }
    //获得最大值，即返回队列首的元素
    int get_max(deque<int> &dq)
    {
        if(!dq.empty())
            return dq.front();
        return {};
    }

    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        deque<int> dq;
        vector<int> result;
        for(int i = 0; i < nums.size(); i++)
        {
            //当窗口还没“建好”时，只添加，不删除
            if(i < k-1)
                push(dq, nums[i]);
            //当窗口刚“建好”时，添加后记录最大值
            else if (i == k-1)
            {
                push(dq, nums[i]);
                result.push_back(get_max(dq));
            }
            //当窗口移动时，先删除再添加后记录
            else
            {
                pop(dq, nums[i-k]);
                push(dq, nums[i]);
                result.push_back(get_max(dq));
            }
        }
        return result;
    }
};

---

LeetCode 347 前k个高频元素 2023.11.6

• 题目链接

• 代码随想录讲解[链接]

• 大小顶堆都类似一个滑动窗口，窗口滑动时，大顶堆是出一个大元素，进一个小元素；小顶堆是出一个小元素，进一个大元素
  

• 题意：输入一个整数数组，输出其中出现频率前k高的元素的一个数组

• 方法：先用map键值对存储数组中出现元素及其出现次数

  • 用大小堆优先队列(滑动窗口)记录频率前k的元素
  • 或对map键值对中的值进行排序，记录前k的元素，参考文章[C++中的map排序链接]

//用哈希表排序的方法
//仿函数，参与vector降序排序
bool cmp(const pair<int, int> a, pair<int, int> b)
{
    return a.second > b.second;
}

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        //创建存放答案的数组
        vector<int> result;
        //因为题目数组长度可能较大，所以用无序map存储每个元素出现的次数
        unordered_map<int, int> unmap;
        for (int i = 0; i < nums.size(); i++)
            unmap[nums[i]]++;
        //测试发现不能对无序map的值进行排序操作，因此选用数组pair数据结构类型进行降序
        //拷贝构造数组
        vector<pair<int, int>> v(unmap.begin(), unmap.end());
        //对数组的键值对中的值进行降序排序
        sort(v.begin(), v.end(), cmp);
        //将出现次数前k的键值对中的键进行记录
        for (int i = 0; i < k; i++)
            result.push_back(v[i].first);
        return result;
    }
};


//大小堆的方法
class Solution {
public:
    class mycomparison{
    public:
        bool operator()(const pair<int, int>& lhs, const pair<int, int> &rhs)
        {
            return lhs.second > rhs.second;
        }
    };

    vector<int> topKFrequent(vector<int>& nums, int k) {
        //创建存放答案的数组
        vector<int> result;
        //因为题目数组长度可能较大，所以用无序map存储每个元素出现的次数
        unordered_map<int, int> unmap;
        for (int i = 0; i < nums.size(); i++)
            unmap[nums[i]]++;
        // 对频率排序
        // 定义一个小顶堆，大小为k
        priority_queue<pair<int, int>, vector<pair<int, int>>, mycomparison> pri_que;
        // 用固定大小为k的小顶堆，扫面所有频率的数值
        for (unordered_map<int, int>::iterator it = unmap.begin(); it != unmap.end(); it++) {
            pri_que.push(*it);
            if (pri_que.size() > k) { // 如果堆的大小大于了K，则队列弹出，保证堆的大小一直为k
                pri_que.pop();
            }
        }
        // 找出前K个高频元素，因为小顶堆先弹出的是最小的，所以倒序来输出到数组
        for (int i = k - 1; i >= 0; i--) {
            result[i] = pri_que.top().first;
            pri_que.pop();
        }
        return result;
    }
};