给你一个整数数组 nums
,请你将数组按照每个值的频率 升序 排序。如果有多个值的频率相同,请你按照数值本身将它们 降序 排序。
请你返回排序后的数组。
示例 1:
输入:nums = [1,1,2,2,2,3]
输出:[3,1,1,2,2,2]
解释:'3' 频率为 1,'1' 频率为 2,'2' 频率为 3 。
示例 2:
输入:nums = [2,3,1,3,2]
输出:[1,3,3,2,2]
解释:'2' 和 '3' 频率都为 2 ,所以它们之间按照数值本身降序排序。
思路:先计入个元素出现的次数(哈希表key为元素大小,val为出现次数),再将哈希表按出现的次数(val)进行排序,出现次数一样则按元素大小(key)排序
class Solution {
public:
vector<int> frequencySort(vector<int>& nums) {
//记录元素出现的次数
unordered_map<int,int> m1;
for(auto e: nums)
{
m1[e]++;
}
//记录出现次数times对应的元素
unordered_map<int,vector<int>> m2;
vector<int> times_v;
for(auto e:m1)
{
if(m2.find(e.second)==m2.end()) times_v.push_back(e.second);
m2[e.second].push_back(e.first);
}
//将元素按出现次数进行排序
sort(times_v.begin(),times_v.end(),less<int>());
//构建返回vector
vector<int> ret;
for(int i=0;i<times_v.size();i++)
{
int j=times_v[i];
vector<int> k=m2[times_v[i]];
sort(k.begin(),k.end(),greater<int>());
for(auto e:k)
{
for(int n=0;n<j;n++)
{
ret.push_back(e);
}
}
}
return ret;
}
};
答案中是直接对哈希表排序:
class Solution {
public:
vector<int> frequencySort(vector<int>& nums) {
unordered_map<int, int> cnt;
for (int num : nums) {
cnt[num]++;
}
sort(nums.begin(), nums.end(), [&](const int a, const int b) {
if (cnt[a] != cnt[b]) {
return cnt[a] < cnt[b];
}
return a > b;
});
return nums;
}
};
class Solution:
def frequencySort(self, nums: List[int]) -> List[int]:
cnt = Counter(nums)
nums.sort(key=lambda x: (cnt[x], -x))
return nums