quickselect_algorithm.sf 490 B

123456789101112131415161718192021222324
  1. #!/usr/bin/ruby
  2. #
  3. ## https://rosettacode.org/wiki/Quickselect_algorithm
  4. #
  5. func quickselect(a, k) {
  6. var pivot = a.pick;
  7. var left = a.grep{|i| i < pivot};
  8. var right = a.grep{|i| i > pivot};
  9.  
  10. given(var l = left.len) {
  11. when(k) { pivot }
  12. case(k < l) { __FUNC__(left, k) }
  13. default { __FUNC__(right, k - l - 1) }
  14. }
  15. }
  16.  
  17. var v = [9, 8, 7, 6, 5, 0, 1, 2, 3, 4];
  18. var arr = v.range.map{|i| quickselect(v, i)};
  19. say arr;
  20. assert_eq(arr, v.sort);