Rust—如何查找集合中第n个最频繁的元素
我无法想象以前没有人问过这个问题，但我到处都找遍了，找不到答案 我有一个iterable，它包含重复的元素。我想计算每个元素在此iterable中出现的次数，并返回第n个最频繁的元素 我有一个工作代码可以做到这一点，但我真的怀疑这是实现这一点的最佳方式Rust—如何查找集合中第n个最频繁的元素,rust,Rust,我无法想象以前没有人问过这个问题，但我到处都找遍了，找不到答案 我有一个iterable，它包含重复的元素。我想计算每个元素在此iterable中出现的次数，并返回第n个最频繁的元素 我有一个工作代码可以做到这一点，但我真的怀疑这是实现这一点的最佳方式 use std::collections::{BinaryHeap, HashMap}; // returns n-th most frequent element in collection pub fn most_frequent<T
use std::collections::{BinaryHeap, HashMap};
// returns n-th most frequent element in collection
pub fn most_frequent<T: std::hash::Hash + std::cmp::Eq + std::cmp::Ord>(array: &[T], n: u32) -> &T {
// intialize empty hashmap
let mut map = HashMap::new();
// count occurence of each element in iterable and save as (value,count) in hashmap
for value in array {
// taken from https://doc.rust-lang.org/std/collections/struct.HashMap.html#method.entry
// not exactly sure how this works
let counter = map.entry(value).or_insert(0);
*counter += 1;
}
// determine highest frequency of some element in the collection
let mut heap: BinaryHeap<_> = map.values().collect();
let mut max = heap.pop().unwrap();
// get n-th largest value
for _i in 1..n {
max = heap.pop().unwrap();
}
// find that element (get key from value in hashmap)
// taken from https://stackoverflow.com/questions/59401720/how-do-i-find-the-key-for-a-value-in-a-hashmap
map.iter()
.find_map(|(key, &val)| if val == *max { Some(key) } else { None })
.unwrap()
}
使用std:：collections:：{BinaryHeap，HashMap}；
//返回集合中第n个最频繁的元素
发布fn最频繁（数组：&[T]，n:u32）->&T{
//初始化空哈希映射
让mut map=HashMap:：new（）；
//计算iterable中每个元素的出现次数，并在hashmap中另存为（值，计数）
对于数组中的值{
//取自https://doc.rust-lang.org/std/collections/struct.HashMap.html#method.entry
//不太清楚这是怎么回事
让计数器=映射项（值）或插入（0）；
*计数器+=1；
}
//确定集合中某些元素的最高频率
让mut heap:BinaryHeap=map.values（）.collect（）；
让mut max=heap.pop（）.unwrap（）；
//获取第n个最大值
对于1..n中的i{
max=heap.pop（）.unwrap（）；
}
//查找该元素（从hashmap中的值获取键）
//取自https://stackoverflow.com/questions/59401720/how-do-i-find-the-key-for-a-value-in-a-hashmap
map.iter（）
.find_map（|（key，&val）|如果val==max{Some（key）}否则{None}）
.unwrap（）
}
有没有更好的方法或更优化的
stduse std::cmp::{Eq, Ord, Reverse};
use std::collections::{BinaryHeap, HashMap};
use std::hash::Hash;
pub fn most_frequent<T>(array: &[T], k: usize) -> Vec<(usize, &T)>
where
T: Hash + Eq + Ord,
{
let mut map = HashMap::new();
for x in array {
*map.entry(x).or_default() += 1;
}
let mut heap = BinaryHeap::with_capacity(k + 1);
for (x, count) in map.into_iter() {
heap.push(Reverse((count, x)));
if heap.len() > k {
heap.pop();
}
}
heap.into_sorted_vec().into_iter().map(|r| r.0).collect()
}
使用std:：cmp:：{Eq，Ord，Reverse}；
使用std:：collections:：{BinaryHeap，HashMap}；
使用std:：hash:：hash；
发布fn最频繁（数组：&[T]，k:usize）->Vec
哪里
T:散列+Eq+Ord，
{
让mut map=HashMap:：new（）；
对于数组中的x{
*map.entry（x）或_default（）+=1；
}
让mut heap=BinaryHeap:：具有_容量（k+1）；
对于map.into_iter（）中的（x，count）{
push（反向（（计数，x））；
如果heap.len（）>k{
heap.pop（）；
}
}
heap.into_sorted_vec（）.into_iter（）.map（| r | r.0）.collect（）
}
kk[k-1][1]BinaryHeapkkkReverseinto_sorted_vec（）反向此答案中的代码确保堆基本上限于
kknBinaryHeap:：into_sorted_vec（）（Count，item）（usize，&T）t:OrdTOrd