next_permutation.h 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  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_NEXT_PERMUTATION_H
  9. #define _LIBCPP___ALGORITHM_NEXT_PERMUTATION_H
  10. #include <__algorithm/comp.h>
  11. #include <__algorithm/comp_ref_type.h>
  12. #include <__algorithm/iterator_operations.h>
  13. #include <__algorithm/reverse.h>
  14. #include <__config>
  15. #include <__iterator/iterator_traits.h>
  16. #include <__utility/move.h>
  17. #include <__utility/pair.h>
  18. #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
  19. # pragma GCC system_header
  20. #endif
  21. _LIBCPP_PUSH_MACROS
  22. #include <__undef_macros>
  23. _LIBCPP_BEGIN_NAMESPACE_STD
  24. template <class _AlgPolicy, class _Compare, class _BidirectionalIterator, class _Sentinel>
  25. _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 pair<_BidirectionalIterator, bool>
  26. __next_permutation(_BidirectionalIterator __first, _Sentinel __last, _Compare&& __comp) {
  27. using _Result = pair<_BidirectionalIterator, bool>;
  28. _BidirectionalIterator __last_iter = _IterOps<_AlgPolicy>::next(__first, __last);
  29. _BidirectionalIterator __i = __last_iter;
  30. if (__first == __last || __first == --__i)
  31. return _Result(std::move(__last_iter), false);
  32. while (true) {
  33. _BidirectionalIterator __ip1 = __i;
  34. if (__comp(*--__i, *__ip1)) {
  35. _BidirectionalIterator __j = __last_iter;
  36. while (!__comp(*__i, *--__j))
  37. ;
  38. _IterOps<_AlgPolicy>::iter_swap(__i, __j);
  39. std::__reverse<_AlgPolicy>(__ip1, __last_iter);
  40. return _Result(std::move(__last_iter), true);
  41. }
  42. if (__i == __first) {
  43. std::__reverse<_AlgPolicy>(__first, __last_iter);
  44. return _Result(std::move(__last_iter), false);
  45. }
  46. }
  47. }
  48. template <class _BidirectionalIterator, class _Compare>
  49. inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 bool
  50. next_permutation(_BidirectionalIterator __first, _BidirectionalIterator __last, _Compare __comp) {
  51. return std::__next_permutation<_ClassicAlgPolicy>(
  52. std::move(__first), std::move(__last), static_cast<__comp_ref_type<_Compare> >(__comp))
  53. .second;
  54. }
  55. template <class _BidirectionalIterator>
  56. inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 bool
  57. next_permutation(_BidirectionalIterator __first, _BidirectionalIterator __last) {
  58. return std::next_permutation(__first, __last, __less<>());
  59. }
  60. _LIBCPP_END_NAMESPACE_STD
  61. _LIBCPP_POP_MACROS
  62. #endif // _LIBCPP___ALGORITHM_NEXT_PERMUTATION_H