next_permutation.h 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  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_BEGIN_NAMESPACE_STD
  22. template <class _AlgPolicy, class _Compare, class _BidirectionalIterator, class _Sentinel>
  23. _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 pair<_BidirectionalIterator, bool>
  24. __next_permutation(_BidirectionalIterator __first, _Sentinel __last, _Compare&& __comp)
  25. {
  26. using _Result = pair<_BidirectionalIterator, bool>;
  27. _BidirectionalIterator __last_iter = _IterOps<_AlgPolicy>::next(__first, __last);
  28. _BidirectionalIterator __i = __last_iter;
  29. if (__first == __last || __first == --__i)
  30. return _Result(std::move(__last_iter), false);
  31. while (true)
  32. {
  33. _BidirectionalIterator __ip1 = __i;
  34. if (__comp(*--__i, *__ip1))
  35. {
  36. _BidirectionalIterator __j = __last_iter;
  37. while (!__comp(*__i, *--__j))
  38. ;
  39. _IterOps<_AlgPolicy>::iter_swap(__i, __j);
  40. std::__reverse<_AlgPolicy>(__ip1, __last_iter);
  41. return _Result(std::move(__last_iter), true);
  42. }
  43. if (__i == __first)
  44. {
  45. std::__reverse<_AlgPolicy>(__first, __last_iter);
  46. return _Result(std::move(__last_iter), false);
  47. }
  48. }
  49. }
  50. template <class _BidirectionalIterator, class _Compare>
  51. inline _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_SINCE_CXX20
  52. bool
  53. next_permutation(_BidirectionalIterator __first, _BidirectionalIterator __last, _Compare __comp)
  54. {
  55. return std::__next_permutation<_ClassicAlgPolicy>(
  56. std::move(__first), std::move(__last), static_cast<__comp_ref_type<_Compare> >(__comp)).second;
  57. }
  58. template <class _BidirectionalIterator>
  59. inline _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_SINCE_CXX20
  60. bool
  61. next_permutation(_BidirectionalIterator __first, _BidirectionalIterator __last)
  62. {
  63. return _VSTD::next_permutation(__first, __last, __less<>());
  64. }
  65. _LIBCPP_END_NAMESPACE_STD
  66. #endif // _LIBCPP___ALGORITHM_NEXT_PERMUTATION_H