-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathQuickSelect.cs
71 lines (61 loc) · 1.53 KB
/
QuickSelect.cs
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
using System;
namespace Sort
{
public class QuickSelect
{
public static T Find<T>(T[] arr, int k)
where T : IComparable
{
arr = Sort.Shuffle.Run(arr);
k -= 1;
var lo = 0;
var hi = arr.Length - 1;
while (hi > lo)
{
var j = Partition(arr, lo, hi);
if (j < k)
{
lo = j + 1;
}
else if (j > k)
{
hi = j - 1;
}
else
{
return arr[k];
}
}
return arr[k];
}
private static int Partition<T>(T[] arr, int lo, int hi)
where T : IComparable
{
var key = arr[lo];
var i = lo;
var j = hi + 1;
while (true)
{
while (arr[++i].CompareTo(key) < 0)
{
if (i == hi) break;
}
while (arr[--j].CompareTo(key) > 0)
{
if (j == lo) break;
}
if (i >= j) break;
Swap(arr, i, j);
}
Swap(arr, lo, j);
return j;
}
private static void Swap<T>(T[] arr, int from, int to)
where T : IComparable
{
var swap = arr[to];
arr[to] = arr[from];
arr[from] = swap;
}
}
}