aboutsummaryrefslogtreecommitdiff
path: root/src/lib/sort.rs
blob: cff7efce9c28a59194f9db1381600cd8e04b14d2 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import _vec.len;
import _vec.slice;

type lteq[T] = fn(&T a, &T b) -> bool;

fn merge_sort[T](lteq[T] le, vec[T] v) -> vec[T] {

  fn merge[T](lteq[T] le, vec[T] a, vec[T] b) -> vec[T] {
    let vec[T] res = vec();
    let uint a_len = len[T](a);
    let uint a_ix = 0u;
    let uint b_len = len[T](b);
    let uint b_ix = 0u;
    while (a_ix < a_len && b_ix < b_len) {
      if (le(a.(a_ix), b.(b_ix))) {
        res += a.(a_ix);
        a_ix += 1u;
      } else {
        res += b.(b_ix);
        b_ix += 1u;
      }
    }
    res += slice[T](a, a_ix, a_len);
    res += slice[T](b, b_ix, b_len);
    ret res;
  }

  let uint v_len = len[T](v);

  if (v_len <= 1u) {
    ret v;
  }

  let uint mid = v_len / 2u;
  let vec[T] a = slice[T](v, 0u, mid);
  let vec[T] b = slice[T](v, mid, v_len);
  ret merge[T](le,
               merge_sort[T](le, a),
               merge_sort[T](le, b));
}

// Local Variables:
// mode: rust;
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// compile-command: "make -k -C .. 2>&1 | sed -e 's/\\/x\\//x:\\//g'";
// End: