Как сделать функцию, которая выбирает рандомные элементы массива?

Question

[Muranx]

Здравствуйте! Казалось бы банальщина, но не могу додуматься, как сделать функцию, которая будет принимать в качестве параметра количество возвращаемых элементов, и массив, из которого мы будем их вычленять (sorry)!

let arr = [ 1, 2, 3, 4, 5, 6, 7 ];
function getRandom( n, array ) { // n - число рандомных элементов,  из массива array
  // . . . код который позволит получить определённое количество элементов из массива array
};

getRandom( 3, arr ); // должно получиться например [4, 6, 7] или [1, 4, 6]

! ! ! Я написал реализацию данной функции, НО у неё есть минус , который я настоятельно прошу учесть

let arr = [1,2,3,4,5,6,7,8,9,10];

function getRandomRiver(n, a){
    let result = []; 
      for(let k=0; k<a.length; k++){
          if(!Math.round(Math.random())) result.push(a[k]); // Math.round(Math.random()) возвращает либо 0 либо 1, если 0 то мы добавляем этот элемент в массив
          if(result.length>=n) break; 
      };
    console.log(result);
};

getRandomRiver(3, arr);

Дело в том, что данная функция во 1ВЫХ может вернуть и меньшее количество , а во 2ЫХ, она каждый раз рассматривает элементы массива с 0го индекса, и шанс того, что в переменную result попадут элементы ближе к концу массива очень мал, а мне нужно, чтобы был полноценный рандом охватывающий весь массив! Спасибо заранее!

Comments

[Тигран Абрамян]

Хочу уточнить, есть массив из 10-ти чисел, при каждом вызове наша функция из этого массива должна взять рандомно три числа и вывести? и так повторно?

Или при каждом вызове мы из нашего массива который содержит 10 чисел, возьмем три числа и выведем в этом массиве рандомно? какой вариант?

[Muranx]

TAbrahamyan, первый вариант, мне нужно просто взять из этого массива допустим 3 рандомных элемента! Если вам будет так проще, то у меня есть массив див блоков полученных через document.querySelectorAll( ' .div ' ) мне нужно взять n-ное количество рандомных блоков и окрасить их в красный ( ну к примеру ) вот загвоздка в том, что не могу написать функцию, которая бы выбирала рандом из этого массива!

Answer 1

[Иван Клименко]

const arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10];

function getRandomRiver(n, arr) {
    const indexes = new Set();
    const limit = arr.length;
    n = Math.min(n, limit);
    while (indexes.size < n) {
        const index = Math.floor(limit * Math.random());
        indexes.add(index);
    }
    const result = [...indexes].map(index => arr[index]);
    return result;
};

const randomSet = getRandomRiver(3, arr);
console.log(randomSet);

Comments

[Muranx]

хм, благодарю за решение, но я не разбираюсь пока в синтаксисе ES6 поэтому отмечу ваш ответ, НО вернусь к нему позже!

[Иван Клименко]

Muranx, если вы про [...indexes], то это эквивалентно Array.from(indexes).

Если же говорить, почему решение TAbrahamyan неоптимальное, то всё дело в среднем количестве операций, которые будут произведены во время работы алгоритма.

Пусть в исходном массиве будет N элементов. Тогда при сортировке происходит как минимум N · log(N) операций. К этому ещё можно прибавить по N операций на каждый вызов .map().

В итоге, если из массива из 1000 элементов надо выбрать 3 элемента, то будет совершенно примерно 12000 операций.

В предложенном мной варианте количество операций зависит не от количества элементов в массиве, а от количества элементов, которые надо выбрать. И если снова рассмотреть массив из 1000 элементов, из которого надо выбрать 3 элемента, то алгоритм, предложенный мной, затратит всего 12 операций.

[hzzzzl]

Иван Клименко, классный трюк с while (indexes.size < n) на случай когда рандомное число повторится :)

[Иван Клименко]

hzzzzl, да, это именно для этого =)

Answer 2

[Тигран Абрамян]

const getRandomRiver = (n, arr) => arr.sort(() => 0.5 - Math.random()).slice(0, n);

или

const getRandomRiver = (n, arr) => {
  return arr
    .map(x => ({ x, r: Math.random() }))
    .sort((a, b) => a.r - b.r)
    .map(a => a.x)
    .slice(0, n);
}

upd: или держите как хотели

Comments

[Muranx]

Хм, прикольно, перетосовать массив и вернуть первые три элемента, хах! Боже, я себя тупым ощущаю! Второй пример я ещё пока не разобрал, думаю в моём случае и первым можно будет ограничиться! Блин ну как же так то, ну это же блин элементарно !=//

[Иван Клименко]

Muranx, это ОЧЕНЬ неоптимальное решение