set_intersection.h 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  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. #ifndef _LIBCPP___ALGORITHM_SET_INTERSECTION_H
  9. #define _LIBCPP___ALGORITHM_SET_INTERSECTION_H
  10. #include <__algorithm/comp.h>
  11. #include <__algorithm/comp_ref_type.h>
  12. #include <__config>
  13. #include <__iterator/iterator_traits.h>
  14. #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
  15. # pragma GCC system_header
  16. #endif
  17. _LIBCPP_BEGIN_NAMESPACE_STD
  18. template <class _Compare, class _InputIterator1, class _InputIterator2, class _OutputIterator>
  19. _LIBCPP_CONSTEXPR_AFTER_CXX17 _OutputIterator
  20. __set_intersection(_InputIterator1 __first1, _InputIterator1 __last1,
  21. _InputIterator2 __first2, _InputIterator2 __last2, _OutputIterator __result, _Compare __comp)
  22. {
  23. while (__first1 != __last1 && __first2 != __last2)
  24. {
  25. if (__comp(*__first1, *__first2))
  26. ++__first1;
  27. else
  28. {
  29. if (!__comp(*__first2, *__first1))
  30. {
  31. *__result = *__first1;
  32. ++__result;
  33. ++__first1;
  34. }
  35. ++__first2;
  36. }
  37. }
  38. return __result;
  39. }
  40. template <class _InputIterator1, class _InputIterator2, class _OutputIterator, class _Compare>
  41. inline _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_AFTER_CXX17
  42. _OutputIterator
  43. set_intersection(_InputIterator1 __first1, _InputIterator1 __last1,
  44. _InputIterator2 __first2, _InputIterator2 __last2, _OutputIterator __result, _Compare __comp)
  45. {
  46. typedef typename __comp_ref_type<_Compare>::type _Comp_ref;
  47. return _VSTD::__set_intersection<_Comp_ref>(__first1, __last1, __first2, __last2, __result, __comp);
  48. }
  49. template <class _InputIterator1, class _InputIterator2, class _OutputIterator>
  50. inline _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_AFTER_CXX17
  51. _OutputIterator
  52. set_intersection(_InputIterator1 __first1, _InputIterator1 __last1,
  53. _InputIterator2 __first2, _InputIterator2 __last2, _OutputIterator __result)
  54. {
  55. return _VSTD::set_intersection(__first1, __last1, __first2, __last2, __result,
  56. __less<typename iterator_traits<_InputIterator1>::value_type,
  57. typename iterator_traits<_InputIterator2>::value_type>());
  58. }
  59. _LIBCPP_END_NAMESPACE_STD
  60. #endif // _LIBCPP___ALGORITHM_SET_INTERSECTION_H