代码随想录算法训练营Day12：Leetcode239、347

Leetcode239(滑动窗口最大值)：

题目描述：

给你一个整数数组 nums，有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回 滑动窗口中的最大值 。

思路解析：

使用堆（优先队列）来维护当前窗口的最大值，堆存储的是pair类型的元素，是由值本身和对应的下标组成的。比较的是值本身大小，因此我们需要自己定义比较规则，然后初始化窗口，将[0,k-1]这个k个元素加入堆，此时堆顶是当前窗口最大值，然循环滑动窗口，每次滑动判断堆顶元素是否是窗口外的值，如果是则删除，不是，则说明是有效窗口的最大值。

代码及注释:

vector<int> maxSlidingWindow(vector<int>& nums, int k) {

    //定义比较规则
    auto compare = [](pair<int, int>& p1, pair<int, int>& p2) {
        return p1.first < p2.first;
    };

    //定义堆
    priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(compare)> q1(compare);

    vector<int>v1;

    for (int i = 0; i < k; i++) {
        q1.emplace(nums[i], i);
    }
    v1.push_back(q1.top().first);

    for (int i = k; i < nums.size(); i++) {
        q1.emplace(nums[i], i);

        //如果最大值是窗口外的值则需要删除
        while (q1.top().second <= i - k)q1.pop();

        v1.push_back(q1.top().first);
    }
    return v1;

}

Leetcode347（前k个高频元素）：

题目描述：

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

思路解析：

使用unordered_map记录每个元素以及出现的次数，使用priority_queue维护次数最多的元素。

pop priority_queue的k次，存储至vector，完成。

代码及注释:

vector<int> topKFrequent(vector<int>& nums, int k) {
    unordered_map<int, int>umap;

    //哈希表记录出现频率
    for (int i = 0; i < nums.size(); i++) {
        umap[nums[i]]++;
    }

    vector<int>v1;

    auto compare = [](pair<int, int>& p1, pair<int, int>& p2) {
        return p1.second < p2.second;
    };

    priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(compare)> q(compare);

    for (auto& data : umap) {
        q.emplace(data.first, data.second);
    }

    for (int i = 0; i < k; i++) {
        v1.push_back(q.top().first);
        q.pop();
    }
    return v1;
}