mirror of
https://github.com/c3lang/c3c.git
synced 2026-02-27 12:01:16 +00:00
39 lines
1018 B
C
39 lines
1018 B
C
module std::sort;
|
|
|
|
/**
|
|
* Perform a binary search over the sorted array and return the index
|
|
* in [0, array.len) where x would be inserted or cmp(i) is true and cmp(j) is true for j in [i, array.len).
|
|
* @require @is_sortable(list) "The list must be sortable"
|
|
* @require $or(@typeid(cmp) == void*.typeid, @is_cmp_fn(cmp, list)) "Expected a comparison function which compares values"
|
|
**/
|
|
macro usz binarysearch(list, x, cmp = null) @builtin
|
|
{
|
|
usz i;
|
|
usz len = @len_from_list(list);
|
|
for (usz j = len; i < j;)
|
|
{
|
|
usz half = i + (j - i) / 2;
|
|
$if @typeid(cmp) == void*.typeid:
|
|
switch
|
|
{
|
|
case greater(list[half], x): j = half;
|
|
case less(list[half], x): i = half + 1;
|
|
default: return half;
|
|
}
|
|
$else
|
|
$switch
|
|
$case $typeof(cmp).params[0] == @typeid(list[0]):
|
|
int res = cmp(list[half], x);
|
|
$default:
|
|
int res = cmp(&list[half], &x);
|
|
$endswitch
|
|
switch
|
|
{
|
|
case res > 0: j = half;
|
|
case res < 0: i = half + 1;
|
|
default: return half;
|
|
}
|
|
$endif
|
|
}
|
|
return i;
|
|
} |