Data commit

Task/Quickselect-algorithm/Go/quickselect-algorithm-1.go

@@ -0,0 +1,41 @@
+package main
+
+import "fmt"
+
+func quickselect(list []int, k int) int {
+    for {
+        // partition
+        px := len(list) / 2
+        pv := list[px]
+        last := len(list) - 1
+        list[px], list[last] = list[last], list[px]
+        i := 0
+        for j := 0; j < last; j++ {
+            if list[j] < pv {
+                list[i], list[j] = list[j], list[i]
+                i++
+            }
+        }
+        // select
+        if i == k {
+            return pv
+        }
+        if k < i {
+            list = list[:i]
+        } else {
+            list[i], list[last] = list[last], list[i]
+            list = list[i+1:]
+            k -= i + 1
+        }
+    }
+}
+
+func main() {
+    for i := 0; ; i++ {
+        v := []int{9, 8, 7, 6, 5, 0, 1, 2, 3, 4}
+        if i == len(v) {
+            return
+        }
+        fmt.Println(quickselect(v, i))
+    }
+}

Task/Quickselect-algorithm/Go/quickselect-algorithm-2.go

@@ -0,0 +1,55 @@
+package main
+
+import (
+    "fmt"
+    "sort"
+    "math/rand"
+)
+
+func partition(a sort.Interface, first int, last int, pivotIndex int) int {
+    a.Swap(first, pivotIndex) // move it to beginning
+    left := first+1
+    right := last
+    for left <= right {
+        for left <= last && a.Less(left, first) {
+            left++
+        }
+        for right >= first && a.Less(first, right) {
+            right--
+        }
+        if left <= right {
+            a.Swap(left, right)
+            left++
+            right--
+        }
+    }
+    a.Swap(first, right) // swap into right place
+    return right
+}
+
+func quickselect(a sort.Interface, n int) int {
+    first := 0
+    last := a.Len()-1
+    for {
+        pivotIndex := partition(a, first, last,
+	                        rand.Intn(last - first + 1) + first)
+        if n == pivotIndex {
+            return pivotIndex
+        } else if n < pivotIndex {
+            last = pivotIndex-1
+        } else {
+            first = pivotIndex+1
+        }
+    }
+    panic("bad index")
+}
+
+func main() {
+    for i := 0; ; i++ {
+        v := []int{9, 8, 7, 6, 5, 0, 1, 2, 3, 4}
+        if i == len(v) {
+            return
+        }
+        fmt.Println(v[quickselect(sort.IntSlice(v), i)])
+    }
+}