如何在std :: set&lt;中找到最大值pair <int,int =“”>&gt;在给定范围内?

时间:2017-04-09 08:20:06

标签: c++ c++11 stl set

我有一组对,我想找到l和r之间的对的第二个条目中的最大数字。

这就是集合的样子:myset = [(0,2),(1,1),(2,4),(3,0),(4,3)]

以下是我的尝试:

#include <iostream>
#include <set>
using namespace std;

#define INPUT1(x)  scanf("%d", &x)
#define INPUT2(x, y)  scanf("%d%d", &x, &y)
#define OUTPUT1(x) printf("%d\n", x);

bool cmp(pair<int, int> A, pair<int, int> B) {
    return A.second < B.second;
}

int main(int argc, char const *argv[]) {
    int n;
    INPUT1(n);

    set< pair<int,int> > myset;
    set< pair<int,int> >::iterator it;

    for (int i = 0; i < n; i++) {
        int val;
        INPUT(val);
        myset.insert(make_pair(i, val));
    }

    int l, r;
    INPUT2(l, r);
    int max = std::max_element(myset.begin()+l, myset.begin()+r+1, cmp)->second;
    OUTPUT1(max);
}

这不起作用,但是对于l = 1和r = 3,我希望最大值等于4.

我收到以下错误:

invalid operands to binary expression

('iterator' (aka '__tree_const_iterator<std::__1::pair<int, int>, std::__1::__tree_node<std::__1::pair<int, int>, void *> *, long>') and 'int')

4 个答案:

答案 0 :(得分:3)

max_element迭代器返回给最大元素。更不用说集合的元素是对,而不是单个整数。

正确的写作方式是:

int max = std::max_element(myset.begin()+l, myset.begin()+r+1, cmp)->second;

答案 1 :(得分:2)

您不能以这种方式使用std::max_element。原因是std::set提供了双向迭代器,而不是随机访问迭代器,所以禁止使用myset.begin()+l之类的东西。

你应该使用这样的东西:

auto mx = std::numeric_limits<int>::min();

auto first = std::cbegin(myset);
std::advance(first, lf);

auto last = std::cbegin(myset);
std::advance(last, rg + 1);

for (auto it = first; it != std::cend(myset) && it != last; ++it) {
    mx = std::max(mx, it->second);
}

答案 2 :(得分:1)

如果第一个比第二个少,则

比较函数应该返回TRUE 修复

bool cmp(pair<int, int> A, pair<int, int> B) {
    return A.second < B.second;
}

答案 3 :(得分:1)

使用嵌入迭代器的函数对象仍然可以加速搜索maximal:

class Max {
public :
  Max (std::set<std::pair<int, int> >::const_iterator& imax, std::set<std::pair<int, int> >::const_iterator& i) : imax_ (imax), i_ (i), max_ (0) {}
  ~Max () {}
  void operator () (const std::pair<int, int>& p) {
    if (p.second > max_) {
      imax_ = i_;
      max_ = p.second;
    }
    ++i_;
  }
private :
  std::set<std::pair<int, int> >::const_iterator& imax_;
  std::set<std::pair<int, int> >::const_iterator& i_;
  int max_;
};

测试程序:

int main (int argc, char* argv []) {
  std::pair<int, int> tmp [3] = {std::pair<int, int> (2, 1), std::pair<int, int>(3, 4), std::pair<int, int>(5, 2)};
  std::set<std::pair<int, int> > s (tmp, tmp+3);
  size_t l (0), r (3);
  std::set<std::pair<int, int> > ::const_iterator imax (s.begin ()), i(s.begin ()), end (s.end ()), il (s.begin ()), ir (s.begin ());
  std::advance (il, l);
  std::advance (ir, r+1); 
  std::for_each (il, ir, Max (imax, i));
  std::cout << "*imax == (" << imax->first << ", " << imax->second << ")" << std::endl;
}