Методы генерации случайного сочетания — различия между версиями

Материал из Викиконспекты
Перейти к: навигация, поиск
(Псевдокод)
(Псевдокод)
Строка 12: Строка 12:
  
 
===Псевдокод===
 
===Псевдокод===
'''for''' i = 1 '''to''' k  
+
 
  r = rand(1..n);
+
'''for''' i = 1 '''to''' k  
  cur = 0;
+
  r = rand(1..n - i + 1);
  '''for''' j = 1 '''to''' n  
+
  cur = 0;
    '''if''' exist[j]
+
  '''for''' j = 1 '''to''' n  
      cur++;
+
    '''if''' exist[j]
      '''if''' cur == r
+
      cur++;
        insertInAns(a[j]);
+
      '''if''' cur == r
        exist[j] = false;
+
        insertInAns(a[j]);
 +
        exist[j] = false;
 +
 
 +
Здесь <tex>a[]</tex> - исходный массив элементов, <tex>exist[]</tex> - такой массив, что если <tex>exist[i] == 1</tex>, то <tex>i</tex> элемент присутствует в множестве S.
  
 
===Доказательство корректности алгоритма===
 
===Доказательство корректности алгоритма===

Версия 09:05, 27 декабря 2012

Постановка задачи

Необходимо сгенерировать случайное сочетание из [math] n [/math] элементов по [math]k[/math] с равномерным распределением вероятности, если есть в наличии функция для генерации случайного числа в заданном интервале.

Решение за время O(n2)

Пусть S - множество из n элементов, тогда для генерации случайного сочетания сделаем следующее:

  • Выберем в множестве случайный элемент
  • Добавим его в сочетание
  • Удалим элемент из множества

Эту процедуру необъодимо повторить [math]k[/math] раз.

Псевдокод

for i = 1 to k 
  r = rand(1..n - i + 1);
  cur = 0;
  for j = 1 to n 
    if exist[j]
      cur++;
      if cur == r
        insertInAns(a[j]);
        exist[j] = false;

Здесь [math]a[][/math] - исходный массив элементов, [math]exist[][/math] - такой массив, что если [math]exist[i] == 1[/math], то [math]i[/math] элемент присутствует в множестве S.

Доказательство корректности алгоритма

Решение за время O(n)

Для более быстрого решения данной задачи воспользуемся следующим алгоритмом: пусть задан для определенности массив [math]a[][/math] размера [math]n[/math], состоящий из [math]k[/math] единиц и [math]n - k[/math] нулей. Применим к нему алгоритм генерации случайной перестановки. Тогда все элементы [math]i[/math], для которых [math]a[i] = 1[/math], включим в сочетание.

Псевдокод

 for i = 1 to n 
   if i <= k
     a[i] = 1;
   else
     a[i] = 0;
 random_shuffle(a);
 for i = 1 to n
   if a[i] == 1
     insertInAnswer(i);

Доказательство корректности алгоритма

Оценка временной сложности

Заметим, что алгоритм состоит из 2 невложенных циклов по [math]n[/math] итераций каждый и функции генерации случайной перестановки [math]random\_shuffle()[/math], работающей за [math]O(n)[/math] по алгоритму Фишера Йетcа. Следовательно, временная сложность и всего алгоритма [math]O(n)[/math]

См. также

Источники