347-前k个高频元素

LeetCode 347. Top K Frequent Elements

LeetCode-347

Given an integer array nums and an integer k, return the k most frequent elements. You may return the answer in any order.

Example 1:

Input: nums = [1,1,1,2,2,3], k = 2
Output: [1,2]

method: 小根堆

类似于215-数组中的第k个最大元素

超过k个就弹出最小的，相当于排序之后砍掉前面size-k个，留下的就是最大的k个

哈希表存储每个数字出现次数
用pair<int,int>存储数值和出现的频率，所以要自定义优先队列的排序方式

class cmp {
public:
    bool operator()(const pair<int,int>& lhs, const pair<int,int>& rhs) {
        return lhs.second > rhs.second; // 跟sort不一样，反过来了
    }
};
vector<int> topKFrequent(vector<int>& nums, int k) {
    unordered_map<int, int> hash;
    for (auto n : nums) hash[n]++;
    priority_queue<pair<int,int>, vector<pair<int,int>>, cmp> q;
    for (auto it = hash.begin(); it != hash.end(); it++) {
        q.push(*it);
        if (q.size() > k) q.pop();  // 大于k个要把最小的出队
    } 
    vector<int> res;
    while (!q.empty()) {
        res.push_back(q.top().first);
        q.pop();
    }
    return res;
}

时间复杂度：$O(Nlogk)$，堆的大小至多为$k$
空间复杂度：$O(N)$

---

LeetCode 692. 前K个高频单词

给定一个单词列表 words 和一个整数 k ，返回前 k 个出现次数最多的单词。

返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率， 按字典顺序 排序。

示例 1：

输入: words = ["i", "love", "leetcode", "i", "love", "coding"], k = 2
输出: ["i", "love"]
解析: "i" 和 "love" 为出现次数最多的两个单词，均为2次。
    注意，按字母顺序 "i" 在 "love" 之前。

method

方法是一样的，只是频率相同时要按字典序排

堆的自定义：想让元素从小到大排序，就return lhs>rhs，想让元素从大到小排序，就return lhs<rhs

class cmp {
public:
    bool operator()(const pair<string, int>& lhs, const pair<string, int>& rhs) {
        if (lhs.second != rhs.second) return lhs.second > rhs.second;
        return lhs.first < rhs.first;	// 频率相同要按字典序从大到小排，后面的权重更大
    }
};
class Solution {
public:
    vector<string> topKFrequent(vector<string>& words, int k) {
        priority_queue<pair<string,int>, vector<pair<string,int>>, cmp> q;
        unordered_map<string, int> hash;
        for (auto w : words) {
            hash[w]++;
        }
        for (auto it = hash.begin(); it != hash.end(); it++) {
            q.push(*it);
            if (q.size() > k) q.pop();
        }
        vector<string> res(q.size());
        for (int i = res.size() - 1; i >= 0; i--) {
            res[i] = q.top().first;
            q.pop();
        }
        return res;
    }
};