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
operator ()Comp24 bool operator()(const std::pair<int, int>& lhs,
25 const std::pair<int, int>& rhs) const {
26 return lhs < rhs;
27 }
28
operator ()Comp29 bool operator()(const std::pair<int, int>& lhs, int rhs) const {
30 return lhs.first < rhs;
31 }
32
operator ()Comp33 bool operator()(int lhs, const std::pair<int, int>& rhs) const {
34 return lhs < rhs.first;
35 }
36 };
37
38 template <typename Container>
test()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
main(int,char **)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