Можно ли сделать функцию универсальной?

Question

Решил для тренировки написать функцию бинарного поиска целых чисел в массиве. Есть два массива и функция:

let arr1 = [1, 2, 3, 4, 5, 6, 7, 8, 9];
let arr2 = [9, 8, 7, 6, 5, 4, 3, 2, 1];

let binarySerch = (arr, num) => {

    let currentIndex = arr.length / 2;

    if (arr[0] == num) return 0;

    while (arr[currentIndex] != num) {
        if (arr[currentIndex] < num) currentIndex /= 2;
        else currentIndex *= 2;
    }
    if (currentIndex == 1 && arr[0] != num) return -1;
    return currentIndex;
}

Эта функция по идее должна работать для массива сортированного по убыванию. Можно ли как-то адаптировать ее под массив сортированный по возрастанию тоже, но не дублировать код? Например, написать что-то типа:

if(arr[arr.length -1] < arr[arr.length -2]) // сортирован по убыванию и надо двигаться в одном направлении, иначе - в противоположном

Answer 1

Для массива отсортированного по возрастанию достаточно проверять противоположное условие, когда решаете в какой из половин нужно продолжать поиск, т.е. проверять условие arr[currentIndex] > num.

Чтоб совместить это в одной функции, можно передать какой именно результат сравнения мы ожидаем:

let binarySerch = (arr, num, sort_descending = true) => {

    let currentIndex = arr.length / 2;

    if (arr[0] == num) return 0;

    while (arr[currentIndex] != num) {
        if ((arr[currentIndex] < num) == sort_descending) currentIndex /= 2;
        else currentIndex *= 2;
    }
    if (currentIndex == 1 && arr[0] != num) return -1;
    return currentIndex;
}

Тогда для сортировки по убыванию вызов будет как и раньше:

binarySearch(arr, num)

или

binarySearch(arr, num, true)

Для отсортированного по возрастанию вызываем так:

binarySearch(arr, num, false)

Answer 2

Если поработать напильником, то можно сделать так.

let arr1 = [1, 2, 3, 4, 5, 6, 7, 8, 9];
let arr2 = [9, 8, 7, 6, 5, 4, 3, 2, 1];


let binarySearch = function(arr, num, order="asc"){
   
    let start=0
    let end=arr.length-1;
    
    while (start<=end){
    
        let mid=((start + end)/2) | 0;
        if (arr[mid]=== num) return mid;
        else if ((arr[mid] < num) == (order == "asc")) 
             start = mid + 1;
        else
             end = mid - 1;
    }
    
    return -1;
}

console.log(binarySearch(arr1, 2));
console.log(binarySearch(arr2, 2, "desc"));

БЛОГ НА HUSL

Можно ли сделать функцию универсальной?

Ответы (2 шт):