v / examples / quick_sort.v
41 lines · 36 sloc · 942 bytes · f09826e928f9612bab9299faefff7cf34a503362
Raw
1import rand
2
3const gen_len = 1000 // how many random numbers to generate
4
5const gen_max = 10000
6
7fn main() {
8 mut arr := []int{}
9 for _ in 0 .. gen_len {
10 arr << rand.intn(gen_max) or { 0 }
11 }
12 println('length of random array is ${arr.len}')
13 println('before quick sort whether array is sorted: ${is_sorted[int](arr)}')
14 quick_sort[int](mut arr, 0, arr.len - 1)
15 println('after quick sort whether array is sorted: ${is_sorted[int](arr)}')
16}
17
18fn quick_sort[T](mut arr []T, l int, r int) {
19 if l >= r {
20 return
21 }
22 mut sep := l // what is sep: [...all_value<arr[sep]...sep...all_value>=arr[sep]...]
23 for i in l + 1 .. r + 1 {
24 if arr[i] < arr[l] {
25 sep++
26 arr[i], arr[sep] = arr[sep], arr[i]
27 }
28 }
29 arr[l], arr[sep] = arr[sep], arr[l]
30 quick_sort[T](mut arr, l, sep - 1)
31 quick_sort[T](mut arr, sep + 1, r)
32}
33
34fn is_sorted[T](arr []T) bool {
35 for i in 0 .. arr.len - 1 {
36 if arr[i] > arr[i + 1] {
37 return false
38 }
39 }
40 return true
41}
42