之前没有自己写过桶排序的代码,这次遇到总结一下。
//桶排序
class Solution {
public:
string frequencySort(string s) {
unordered_map<char, int> m;
for (char c : s)
m[c]++;
vector<string> buckets(s.size()+1, "");
for (auto& item : m) {
buckets[item.second].append(item.second, item.first);
}
string ret = "";
for (int i = buckets.size()-1; i > 0; --i)
if (!buckets[i].empty())
ret.append(buckets[i]);
return ret;
}
};
另外自己还有两个方法,记得去submission里面看看。
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int, int> m;
for (int num : nums)
m[num]++;
int sz = nums.size();
vector<vector<int>> buckets(sz+1, vector<int>());
for (auto& item : m)
buckets[item.second].push_back(item.first);
vector<int> ret;
for (int i = sz; i >= 0; --i) {
if (!buckets[i].empty()) {
for (int j = 0; j < buckets[i].size(); ++j) {
ret.push_back(buckets[i][j]);
if (--k == 0)
return ret;
}
}
}
return ret;
}
};
class Solution {
public:
vector<string> topKFrequent(vector<string>& words, int k) {
unordered_map<string, int> m;
for (auto& word : words)
m[word]++;
int sz = words.size();
vector<vector<string>> buckets(sz+1, vector<string>());
for (auto& item : m)
buckets[item.second].push_back(item.first);
vector<string> ret;
for (int i = sz; i >= 0; --i) {
if (!buckets[i].empty()) {
sort(buckets[i].begin(), buckets[i].end());
for (int j = 0; j < buckets[i].size(); ++j) {
ret.push_back(std::move(buckets[i][j]));
if (--k == 0)
return std::move(ret);
}
}
}
return std::move(ret);
}
};
总的来说,我对桶排序的看法:
1.首先需要声明包含所有可能情况的数量的桶。比如算出现最多次数的元素,每个位置的桶代表出现相应次数的元素的集合。那么就需要从1-元素总数这么多个桶,因为需要包含所有情况。确实有点浪费空间。
2.将对应元素放进相应桶的过程,可以看成是映射,其实感觉来说更像hash表的hash映射过程,放到对应的位置上。
3.桶之间的顺序是定了,然后桶内的次序记得也要关注,比如上面第三个,桶内也要有次序。