llvm-project

1
//===----------------------------------------------------------------------===//
2
//
3
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4
// See https://llvm.org/LICENSE.txt for license information.
5
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6
//
7
//===----------------------------------------------------------------------===//
8

9
// UNSUPPORTED: c++03, c++11, c++14, c++17
10

11
// <set>
12

13
// class set
14

15
// template<typename K> bool contains(const K& x) const; // C++20
16

17
#include <cassert>
18
#include <set>
19
#include <utility>
20

21
struct Comp {
22
  using is_transparent = void;
23

24
  bool operator()(const std::pair<int, int>& lhs,
25
                  const std::pair<int, int>& rhs) const {
26
    return lhs < rhs;
27
  }
28

29
  bool operator()(const std::pair<int, int>& lhs, int rhs) const {
30
    return lhs.first < rhs;
31
  }
32

33
  bool operator()(int lhs, const std::pair<int, int>& rhs) const {
34
    return lhs < rhs.first;
35
  }
36
};
37

38
template <typename Container>
39
void test() {
40
  Container s{{2, 1}, {1, 2}, {1, 3}, {1, 4}, {2, 2}};
41

42
  assert(s.contains(1));
43
  assert(!s.contains(-1));
44
}
45

46
int main(int, char**) {
47
  test<std::set<std::pair<int, int>, Comp> >();
48
  test<std::multiset<std::pair<int, int>, Comp> >();
49

50
  return 0;
51
}
52