C ++查找地图的第N个最高元素

时间:2017-02-11 13:10:50

标签: c++ sorting vector hashmap

我有一个字符串向量std::vector<string> list,我试图找到向量的第N个最高重复元素。

我有一张地图包含了矢量元素和重复数字。

std::map<std::string , int> mapa;
for(int i = 0 ; i<list.size() ; i++)
  mapa[list[i]]++;

如何从地图中找到第N个最高的?

示例矢量:

qwe asd qwe asd zxc asd zxc qwe qwe asd sdf asd fsd 

如果N是2,我需要像

那样的输出
asd 5
qwe 4

1 个答案:

答案 0 :(得分:3)

您可以使用std::partial_sort

std::map<std::string, std::size_t>
compute_frequencies(const std::vector<std::string>& words)
{
    std::map<std::string, std::size_t> res;
    for(const auto& word : words) {
        res[word]++;
    }
    return res;    
}

std::vector<std::pair<std::string, std::size_t>>
as_vector(const std::map<std::string, std::size_t>& m)
{
    return {m.begin(), m.end()};
}

int main() {
    const std::vector<std::string> words{
        "qwe", "asd", "qwe", "asd", "zxc", "asd",
        "zxc", "qwe", "qwe", "asd", "sdf", "asd", "fsd"
    };
    auto frequencies = as_vector(compute_frequencies(words));
    std::partial_sort(frequencies.begin(), frequencies.end(), frequencies.begin() + 2,
        [](const auto& lhs, const auto& rhs) {
            return lhs.second > rhs.second;    
        });
    for (std::size_t i = 0; i != 2; ++i) {
        std::cout << frequencies[i].first << " " << frequencies[i].second << std::endl;  
    }
}

Demo