Получение объекта по номеру

Описание алгоритма

Получаем элементы объекта по порядку: сначала определим, какой элемент будет стоять на первом месте, потом на втором и так далее. Считаем, что мы нашли первые [math]i[/math] элементов объекта. Для всех вариантов элемента, который может стоять на позиции с номером [math]i+1[/math], посчитаем диапазон номеров, который будет соответствовать объектам с данным префиксом. Если искомый номер входит в один из диапазонов, то, очевидно, мы нашли элемент, который должен стоять на месте с номером [math]i+1[/math]. Диапазоны номеров не пересекаются, значит на это место больше нельзя поставить никакой другой элемент:

• в начале каждого шага [math]\mathtt{numOfObject}[/math] — номер нужного объекта среди тех, у которых префикс до [math]i[/math]-го элемента лексикографически равен префиксу нашего объекта,
• [math]\mathtt{n}[/math] — количество мест в комбинаторном объекте (например, битовый вектор длины [math]n[/math]),
• [math]\mathtt{k}[/math] — количество различных элементов, которые могут находиться в данном комбинаторном объекте. Например, для битового вектора [math]k=2[/math], поскольку возможны только [math]0[/math] и [math]1[/math]. Все элементы занумерованы в лексикографическом порядке, начиная с [math]1[/math].

Комбинаторные объекты занумерованы с [math]0[/math]. Переход к нумерации с единицы можно сделать с помощью одной операции декремента перед проходом алгоритма:

function num2object(numOfObject: int):
  for i = 1 to n                     
    for j = 1 to k                      
      if j-й элемент можно поставить на i-e место 
        if numOfObject >= (количество комбинаторных объектов с префиксом object[1..i-1] и элементом j на месте i)
          numOfObject -= (количество комбинаторных объектов с префиксом object[1..i-1] и элементом j на месте i)
        else
          object[i] = j
          break
  return object

Сложность алгоритма — [math]O(nk) [/math]. Количества комбинаторных объектов с заданными префиксами считаются известными, и их подсчет в сложности не учитывается. Стоит отметить, что подсчет количества комбинаторных объектов с заданным префиксом зачастую является задачей с достаточно большой вычислительной сложностью. Приведем примеры получения некоторых комбинаторных объектов по номеру.

Битовые вектора

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

• [math]\mathtt{bitvector[n]}[/math] — искомый битовый вектор,
• [math]\mathtt{numOfBitvector}[/math] — номер искомого вектора среди всех битовых векторов,
• [math]\mathtt{pow(2, n)}[/math] — [math]2^{n}[/math] количество битовых векторов длины [math]n[/math],

vector<int> num2bitvector(numOfBitvector: int):
  for i = 1 to n                                      
   if numOfBitvector >= pow(2, (n - i))
     numOfBitvector -= pow(2, (n - i))
     bitvector[i] = 1
   else
     bitvector[i] = 0    
  return bitvector    

Данный алгоритм работает за [math]O(n)[/math], так как в случае битовых векторов [math]k[/math] не зависит от [math]n[/math]. Алгоритм эквивалентен переводу числа из десятичной системы в двоичную.

Перестановки

Рассмотрим алгоритм получения [math]k[/math]-ой в лексикографическом порядке перестановки размера [math]n[/math]. Заметим, что всем префиксам на каждом шаге будет соответствовать диапазон номеров одинакового размера, (так как количество всевозможных суффиксов зависит только от длины) то есть можем просто посчитать "количество диапазонов, которые идут до нас" (количество цифр уже полностью занятых перестановками с меньшим номером) за [math]O(1) [/math]:

• [math]\mathtt{k}[/math] — номер искомой последовательности,
• [math]\mathtt{n!}[/math] — количество перестановок размера [math]n[/math],
• [math]\mathtt{permutation[n]}[/math] — искомая перестановка,
• [math]\mathtt{was[n]}[/math] — использовали ли мы уже эту цифру в перестановке.

На [math]i[/math]-ом шаге:

• [math]\mathtt{alreadyWas}[/math] — сколько цифр уже полностью заняты перестановками с меньшим номером,
• мы должны поставить ту цифру, которая еще полностью не занята, то есть цифру с номером [math]alreadyWas + 1[/math]. Среди цифр, которых еще нет в нашем префиксе, считаем, что это цифра [math]j[/math].

На [math]j[/math]-ом шаге:

• [math]\mathtt{curFree}[/math] — если элемент с номером [math]j[/math] свободен, то он имеет номер curFree среди всех свободных элементов с [math]1[/math] по [math]j[/math].

list<int> num2permutation(k: int):
  for i = 1 to n                               
    alreadyWas = k / (n - i)!      
    k %= (n - i)!
    curFree = 0
    for j = 1 to n  
      if was[j] == false 
        curFree++
        if curFree == alreadyWas + 1
          permutation[i] = j
          was[j] = true
  return permutation

Данный алгоритм работает за [math]O(n^2)[/math], так как в случае перестановок [math]n=k[/math]. Мы можем посчитать все [math]\mathtt{n!}[/math] за [math]O(n) [/math]. Асимптотику можно улучшить до [math]O(n \log {n}) [/math], если использовать структуры данных (например, декартово дерево по неявному ключу), которые позволяют искать [math]i[/math]-й элемент множества и удалять элемент множества за [math]O( \log {n}) [/math].

Сочетания

На каждой итерации мы проверяем, входит ли число [math]\mathtt{next}[/math] в искомое сочетание. Если мы хотим взять [math]\mathtt{next}[/math], то номер сочетания должен быть меньше, чем [math]\binom{n - 1}{k - 1}[/math], так как потом надо будет выбрать [math]k - 1[/math] элемент из [math]n - 1[/math] доступных. Если нет, то будем считать, что [math]\binom{n - 1}{k - 1}[/math] сочетаний, начинающихся с [math]\mathtt{next}[/math], мы пропустили. В обоих случаях рассмотрение текущего числа [math]next[/math] мы заканчиваем и переходим к следующему числу.

• [math]\mathtt{choose}[/math] — искомое сочетание,
• [math]\mathtt{C[n][k]}[/math] — количество сочетаний из [math]n[/math] по [math]k[/math], [math]\mathtt{C[n][0] = 1}[/math],

list<int> num2choose(n, k, m: int):
  next = 1
  while k > 0
    if m < C[n - 1][k - 1]
      choose.push_back(next)
      k = k - 1
    else
      m -= C[n - 1][k - 1]
    n = n - 1
    next = next + 1
  return choose

Асимптотика приведенного алгоритма — [math]O(n)[/math], предподсчет [math]\mathtt{C[n][k]}[/math] — [math]O(n^2)[/math]

См. также

• Получение номера по объекту
• Получение предыдущего сочетания
• Генерация следующего сочетания

Источники информации

• Программирование в алгоритмах / С. М. Окулов. — М.: БИНОМ. Лаборатория знаний, 2002. стр.31 - ISBN 5-94774-010-9