random_shuffle_urng.pass.cpp 1.2 KB

123456789101112131415161718192021222324252627282930313233343536
  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. // <algorithm>
  9. // template<class RandomAccessIterator, class UniformRandomNumberGenerator>
  10. // void shuffle(RandomAccessIterator first, RandomAccessIterator last,
  11. // UniformRandomNumberGenerator& g);
  12. #include <algorithm>
  13. #include <random>
  14. #include <cassert>
  15. #include "test_macros.h"
  16. int main(int, char**)
  17. {
  18. int ia[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
  19. int ia1[] = {2, 7, 1, 4, 3, 6, 5, 10, 9, 8};
  20. int ia2[] = {1, 8, 3, 4, 6, 9, 5, 7, 2, 10};
  21. const unsigned sa = sizeof(ia)/sizeof(ia[0]);
  22. std::minstd_rand g;
  23. std::shuffle(ia, ia+sa, g);
  24. LIBCPP_ASSERT(std::equal(ia, ia+sa, ia1));
  25. assert(std::is_permutation(ia, ia+sa, ia1));
  26. std::shuffle(ia, ia+sa, std::move(g));
  27. LIBCPP_ASSERT(std::equal(ia, ia+sa, ia2));
  28. assert(std::is_permutation(ia, ia+sa, ia2));
  29. return 0;
  30. }