在多地图

时间:2018-04-24 14:15:26

标签: c++ key multimap std-pair

我有一个由一对密钥组成的多图,该密钥是一对整数,而密钥的值是浮点数。 现在我想拥有所有键的2个最大值 在我的例子中,Key(1,1)应该产生值5.8和3.7 键(2,2)应该产生2.4和1.5 指出:我不知道我的钥匙的数量和外观。所以我不知道密钥(2,2)是否存在。

以下是代码:

int main(int argc, char** argv)
{
    // Create some values - both vectors have the same size
    // The pairs and the thetas may be unsorted
    vector<pair<int, int>> pairs;
    pairs.push_back(make_pair(1, 1));
    pairs.push_back(make_pair(1, 1));
    pairs.push_back(make_pair(1, 1));
    pairs.push_back(make_pair(2, 2));
    pairs.push_back(make_pair(2, 2));
    pairs.push_back(make_pair(3, 3));
    pairs.push_back(make_pair(3, 3));
    pairs.push_back(make_pair(1, 1));

    vector<float> theta;
    theta.push_back(1.4);
    theta.push_back(2.4);
    theta.push_back(3.7);
    theta.push_back(2.4);
    theta.push_back(1.5);
    theta.push_back(1.6);
    theta.push_back(2.4);
    theta.push_back(5.8);

    multimap<pair<int, int>, float> similarities;

    for (size_t i = 0; i < pairs.size(); i++) {
        similarities.insert(make_pair(make_pair(pairs[i].first, pairs[i].second), theta[i]));
    }

}

在我的具体情况下,我不知道在我的多图中定义了哪些密钥对。 我认为也许多图不是正确的选择,但我不确定什么是更好的类型 我试图保持theta的矢量格式,但是很难跟踪相关的对。

2 个答案:

答案 0 :(得分:1)

我希望我帮到你:

auto range = similarities.equal_range(make_pair(1,1));

size_t found = 0; float highest, second_highest;

if (range.first != similarities.end() && range.first != range.second) {
  ++found;
  highest = (*std::max_element(range.first++, range.second)).second;
  if (range.first != range.second) {
    ++found;
    second_highest =
      (*std::max_element(range.first++, range.second, [&](auto a, auto b){ return a.second < b.second && b.second < highest; })).second;
  }
}

if (found > 0)
  std::cout << highest<< std::endl;
if (found > 1)
  std::cout << second_highest << std::endl;

上面的内容不会给你两次相同的值。

答案 1 :(得分:1)

你有几个选择。

  • 保留现有的multimap,然后对值进行线性搜索,找到最大的两个。
  • 改为使用map<pair<int, int>, set<float, std::greater<float> > >,然后在查找密钥后立即可以访问前两项。
  • 使用map<pair<int, int>, vector<float> >并在插入时对矢量进行排序。在执行更多查找而不是插入时更好。
  • 在需要检索最高两个值时,使用map<pair<int, int>, vector<float> >partial_sort前两项。在执行更多插入而不是查找时更好。