При вводе элемента в дерево случайного поиска этому элементу приcваивается приоритет — некоторое вещественное число с равномерным распределением в диапазоне [0, 1] (вероятность выбора любого числа в заданном диапазоне одинакова). Приоритеты элементов в дереве случайного поиска определяют их положение в дереве в соответствии с правилом: приоритет каждого элемента в дереве не должен быть более приоритета любого из его последователей. Правило двоичного дерева поиска также остается справедливым: для каждого элемента х элементы в левом поддереве х будут меньше, чем х, а в правом поддереве — больше, чем х. На рис. 1 приведен пример такого дерева случайного поиска.

Рис. 1: Дерево случайного поиска. Приоритеты каждого элемента указаны в виде нижнего индекса

Как же осуществить ввод некоторого элемента х в дерево случайного поиска? Он выполняется в два этапа. На первом этапе будем учитывать приоритет и введем элемент х уже известным нам способом: проследуем по пути поиска элемента я от корня вниз до внешнего узла, которому принадлежит х, и затем заменим внешний узел новым внутренним узлом, содержащим х. На втором этапе мы изменим форму дерева в соответствии с приоритетами элементов. Сначала случайным образом зададим приоритет для х. В простейшем случае, приоритет х может быть больше и равен приоритету предка, тогда задача оказывается решенной. В противном случае, если приоритет х оказывается меньше приоритета его предка у, то правило приоритета для у оказывается нарушенным. Предположим, что элемент х оказался левым наследником для у. В этом случае применим правую ротацию для элемента у, переместив элемент х в дереве на один уровень выше (рис. 2). Теперь правило приоритета может оказаться нарушенным для нового предка элемента х. Если это так, то применим ротацию для предка — правую ротацию, если х является левым потомком, или левую ротацию, если х является правым потомком, переместив таким образом элемент х еще на один уровень вверх. Такой подъем элемента х будем осуществлять до тех пор, пока он не станет либо корневым, либо его приоритет будет не меньше, чем у его предка.

Рис. 2: Ротация

Рис. 3: Внесение элемента 6 в дерево случайного поиска: (а) и (б)—левая ротация элемента 5; (б) и (в) — правая ротация элемента 7; (в) и (г) - левая ротация элемента 4

На рис. 3 показан процесс внесения элемента. На этой схеме показано, как элемент 6 заменяет соответствующий внешний узел, а затем поднимается вверх в соответствии с его приоритетом (.10) и приоритетами его текущих предков.

Определим теперь соответствующие классы. Несколько позже рассмотрим, как удалить элемент из дерева случайного поиска. Реализация использует классы из описания связанного дерева поиска.

Класс RandomizedNode

Узлы в дереве случайного поиска являются объектами шаблона класса RandomizedNode:

template
class RandomizedNode : public BraidedNode {
 protected :
  RandomizedNode *_parent;
  double _priority;
  void rotateRight (void);
  void rotateLeft (void);
  void bubbleUp (void);
  void bubbleDown (void);
 public :
  RandomizedNode ( Т v, int seed = -1);
  RandomizedNode *lchild (void);
  RandomizedNode *rchild(void);
  RandomizedNode *next (void);
  RandomizedNode *prev{ void);
  RandomizedNode *parent (void);
  double priority (void);
  friend class RandomizedSearchTree;
};

Класс RandomizedNode наследует пять элементов данных из его базового класса: val, _lchild, _rchild, _prev и _next. В класс включены два дополнительных элемента данных: _parent, который указывает на узел-предок текущего узла, и _priority, приоритет текущего узла.

Конструктор присваивает новому узлу RandomizedNode случайное значение приоритета, используя стандартную функцию языка C++ rand, которая генерирует случайное число в пределах от 0 до RAND_MAX :

template
RandomizedNode::RandomizedNode(T v, int seed) :
  BraidedNode(v)
{
  if (seed != -1) srand(seed);
  priority = (rand() % 32767) / 32767.0;
  _parent = NULL;
}

Наибольший интерес представляют компонентные функции класса RandomizedNode, объявленные как private (собственные). Первые две выполняют два вида ротации — правую и левую. Правая ротация является локальной операцией, которая изменяет форму дерева поиска, оставляя неизменным порядок следования элементов — до и после выполнения операции предшественник и последователь для каждого узла остаются без изменения. Правая ротация для узла у приводит к повороту поддерева с корнем в узле у вокруг связи от узла у к его левому потомку х (рис. 2). Операция выполняется путем изменения небольшого числа связей: связь с левым потомком узла у заменяется на связь с поддеревом T2, связь с правым потомком узла х заменяется на связь с узлом у, а связь с узлом у (от его предка) заменяется на связь с узлом х.

Компонентная функция rotateRight выполняет правую ротацию для текущего узла, который играет роль узла у. При этом узел х должен быть левым потомком узла у. Для надежности предполагается, что предок узла у существует, поскольку головной узел должен иметь минимальный приоритет — ни одного другого узла не должно быть на уровне головного узла, поэтому все остальные узлы должны иметь своего предка.

template void RandomizedNode::rotateRight(void)
{
  RandomizedNode *y = this;
  RandomizedNode *x = y->lchild();
  RandomizedNode *p = y->parent();
  y->_lchild = x->rchild();
  if (y->lchild() != NULL)
    y->lchild()->_parent = y;
  if (p->rchild() == y)
    p->_rchild = x;
  else
    p->_lchild = x; 
  x->_parent = p;
  x->_rchild = y;
  y->_parent = x;
}

Действие компонентной функции rotateLeft для левой ротации также показано на рис. 2 и она определяется симметрично:

template void RandomizedNode::rotateLeft(void)
{
  RandomizedNode *х = this;
  RandomizedNode *y = x->rchild();
  RandomizedNode *p = x->parent();
  x->_rchild = y->lchild();
  if (x->rchild() != NULL)
    x->_rchild()->_parent = x;
  if (p->lchild() == x)
    p->_lchild = y;
  else
    p->_rchild = y;
  y->_parent = p;
  y->_lchild = x;
  x->_parent = y;
}

Компонентная функция bubbleUp переносит текущий узел вверх по направлению к корню путем многократных ротаций до тех пор, пока приоритет текущего узла не станет больше или равным, чем у его предка. Функция применяется в тех случаях, когда элемент в дереве случайного поиска располагается ниже, чем это следует в соответствии со значением его приоритета. Отметим, что ротация применяется к предку текущего узла.

template void RandomizedNode::bubbleUp(void)
{
  RandomizedNode *p = parent();
  if (priority() < p->priority() ) {
    if (p->lchild() == this)
      p->rotateRight();
     else
       p->rotateLeft();
     bubbleUp();
  }
}

Компонентная функция bubbleDown переносит текущий узел вниз по направлению к внешним узлам дерева путем многократных ротаций до тех пор, пока приоритет текущего узла не станет меньше или равным приоритетам обоих его потомков. Для ротации ее направление (вправо или влево) зависит от того, какой из потомков текущего узла имеет меньший приоритет. Если, например, приоритет левого потомка меньше, то правая ротация переносит левый потомок вверх на один уровень, а текущий узел перемещается на один уровень вниз. Каждому внешнему узлу приписывается приоритет 2.0, достаточно большой, чтобы ошибочно не переместить его вверх. Функция bubbleDown применяется в тех случаях, когда элемент в дереве случайного поиска располагается выше, чем это следует в соответствии со значением его приоритета. Эта функция будет использована впоследствии для удаления элемента из дерева поиска.

template void RandomizedNode::bubbleDown(void)
{
  float lcPriority = lchild() ? lchild()->priority () : 2.0;
  float rcPriority = rchild() ? rchild()->priority () : 2.0;
  float minPriority = (lcPriority < rcPriority) ? lcPriority : rcPriority;
  if (priority () <= minPriority)
    return;
  if (lcPriority < rcPriority)
    rotateRight();
  else
    rotateLeft();
  bubbleDown();
}

Общедоступные компонентные функции rchild, Ichild, next, prev и parent определяют связи текущего узла с его правым и левым потомками, с последующим и предшествующим узлами и с предком соответственно:

template
RandomizedNode *RandoraizedNode::rchild(void)
{
  return (RandomizedNode*) _rchild;
}

template
RandomizedNode *RandomizedNode::lchild(void)
{
  return (RandomizedNode*) _lchild;
}

template
RandomizedNode *RandomizedNode::next(void)
{
  return ( RandomizedNode*) next;
}

template
RandomizadNode *RandomizedNode::prev(void)
{
  return (RandomizedNode*)_prev;
}

template-Cclass T>
RandomizedNode * RandomizedNode::parent(void)
{
  return (RandcaaizedNode* ) _parent ;
}

Компонентная функция priority возвращает приоритет текущего узла:

template
double RandomizedNode::priority(void)
{
  return _priority;
}

Класс RandomizedSearchTree

Деревья случайного поиска представляются в виде объектов класса RandomizedSearchTree. Во многих отношениях шаблон класса напоминает класс BraidedSearchTree: элемент данных root указывает на головной узел, win представляет окно, а элемент cmp указывает на функцию сравнения деревьев.

template class RandomizedsearchTree {
 private :
  RandomizedNode *root;       // головной узел 
  RandomizedNode *win;        // окно 
  int (*cmp) (T,T);	             // функция сравнения
  void _remove(RandomizedNode*);
 public :
  RandomizedsearchTree (int (*) (T,T) , int = -1);
  ~RandomizedsearchTree (void);
  Т next (void);
  Т prev(void);
  void inorder (void (*) (T) );
  Т val (void);
  bool isFirst (void);
  bool isLast (void);
  bool isHead(void);
  bool isEmpty (void);
  Т find (T);
  Т findMin (void);
  T locate (T);
  T insert (T);
  void remove (void);
  T remove (T);
  T removeMin (void);
};

Конструкторы и деструкторы

Конструктор RandomizedsearchTree инициализирует новое дерево случайного поиска, представляемое изолированным головным узлом с минимальным приоритетом, равным -1,0:

template RandomizedSearchTree::RandomizedsearchTree (int (*с) (Т, Т) , int seed):
   cmp(c)
{
  win = root = new RandomizedNode (NULL, seed); 
  root->_priority = -1.0;
}

Деструктор удаляет дерево поиска:

template
RandomizedSearchTree::~RandomizedsearchTree (void)
{
  delete root;
}

Использование ленты

Компонентные функции next, prev, val, inorder, isFirst, isLast, isHead и is Empty определяются подобно их аналогам в классе

BraidedSearchTree : 
template Т RandomizedSearchTree::next (void)
{
  win = win->next();
  return win->val;
}

template Т RandomizedSearchTree::prev (void)
{
  win = win->prev();
  return win->val;
}

template Т RandomizedSearchTree::val (void)
{
  return win->val;
}

template
void RandomizedSearchTree::inorder (void (*visit) (T) )
{
  RandomizedNode *n = root->next();
  while (n != root) {
    (*visit) (n->val);
    n = n->next();
  }
}

template
bool RandomizedSearchTree::isFirst (void)
{
  return (win == root->next() ) && (root != root->next ( ) );
}

template
bool RandomizedSearchTree::isLast (void)
{
  return (win == root->prev() ) && (root != root->next() );
}

template
bool RandomizedSearchTree::isHead (void)
{
  return (win == root);
}

template
bool RandomizedSearchTree::isEmpty (void)
{
  return (root == root->next() );
}

Поиск

Компонентные функции find и findMin применяются также подобно их аналогам в классе BraidedSearchTree :

template Т RandomizedSearchTree::find (T val)
{
  RandomizedNode *n = root->rchild ( );
  while (n) {
    int result = (*cmp) (val, n->val);
    if (result < 0)
      n = n->lchild();
    else if (result > 0)
      n = n->rchild();
    else {
      win = n;
      return n->val;
    }
  }
  return NULL;
}

template Т RandomizedSearchTree::findMin (void)
{
  win = root->next();
  return win->val;
}

Теперь мы введем новую операцию поиска locate. При обращении к функции locate с аргументом val возвращается наибольший элемент в дереве, который не больше, чем значение val. Если в дереве присутствует элемент со значением val, то возвращается значение val. Если такого значения нет, то возвращается значение меньше, чем val, если же в дереве и такого значения нет, то возвращается NULL.

Для выполнения этой операции будем отслеживать путь по дереву поиска до значения val. По мере продвижения вниз запоминается последний (самый нижний) узел b, в котором поворачиваем вправо — узел b является самым нижним узлом, обнаруженным на пути поиска, для которого правый потомок также лежит на пути поиска. Если в дереве находится значение val, то оно просто возвращается. В противном случае, если мы не найдем значение val в дереве — путь поиска завершается на внешнем узле — то возвращается элемент, запомненный в узле b.

Операция реализуется компонентной функцией locate, в которой окно перемещается на обнаруженный элемент и возвращается этот элемент.

template Т RandomizedSearchTree::locate (T val)
{
  RandomizedNode *b = root;
  RandomizedNode *n = root->rchild();
  while (n) {
    int result = (*cmp) (val, n->val);
    if (result < 0)
      n = n->lchild();
    else if (result > 0) {
      b = n;
      n = n->rchild();
    } else {
      win = n;
      return win->val;
    }
  }
  win = b;
  return win->val;
}

Каким же образом получается правильный результат? Он вполне очевиден, если значение val точно находится в дереве. Но предположим, что такого значения в дереве нет, поэтому поиск значения val завершается на внешнем узле. Поскольку указатель n сигнализирует об окончании поиска, значение b показывает на наибольший элемент, который меньше всех остальных, содержащихся в поддереве с корнем в узле n. Легко видеть, что это условие сохраняется всегда. Каждый раз, когда в процессе поиска в узле n мы переходим на левую ветвь, условие продолжает сохраняться, поскольку каждый элемент в левом поддереве узла n меньше элемента в узле n. При переходе в узле n на правую ветвь присваивание b значения, равного n, восстанавливает условие, поскольку: (1) каждый элемент в правом поддереве узла n больше элемента, хранящегося в n и (2) элемент, хранящийся в узле n больше, чем элемент, на который указывало значение b до его изменения. И, наконец, когда n указывает на внешний узел, то b будет наибольшим элементом в дереве, который меньше любого другого элемента, который может легально заменить внешний узел, среди которых находится также и val.

Включение элементов

Для внесения нового элемента в дерево случайного поиска сначала найдем внешний узел, к которому он принадлежит (этап 1), а затем будем его поднимать по дереву по направлению к корню в соответствии с его приоритетом (этап 2). Компонентная функция insert заносит элемент val в текущее дерево случайного поиска и затем перемещает окно на этот элемент и возвращает указатель на этот элемент:

template T RandomizedSearchTree::insert (T val)
{
                                          // этап 1
  int result = 1;
  RandomizedNode *p = root;
  RandomizedNode *n = root->rchild();
  while (n) {
    p = n;
    result = (*cmp) (val, p->val);
    if (result < 0)
      n = p->lchild();
    else if (result > 0)
      n = p->rchild();
    else
      return NULL;
  }
  win = new RandomizedNode(val);
  win->_parent = p;
  if (result < 0) {
    p->_lchild = win;
    p->prev()->Node::insert(win);
  } else {
    p->_rchild = win;
    p->Node::insert(win);
  }
                                          //  этап 2
  win->bubbleUp ();
  return val;
}

Удаление элементов

Компонентная функция remove удаляет элемент в окне и затем перемещает окно в предшествующую позицию. Компонентная функция removeMin (удаляет наименьший элемент и возвращает его. Если окно находилось на этом элементе, то оно перемещается в головную позицию:

template void RandomizedSearchTree::remove(void)
{
  if (win != root)
    _remove(win);
}

template Т RandomizedSearchTree::removeMin(void)
{
  Т val = root->next()->val;
  if (root != root->next() )
    _remove(root->next());
  return val;
}

Обе эти функции основаны на внутренней компонентной функции _remove. Для удаления узла n функция _remove увеличивает приоритет узла n до 1.5 и затем опускает его вниз до тех пор, пока он не станет внешним узлом. Приоритет 1.5 превышает приоритет любого элемента в дереве, но меньше, чем приоритет внешних узлов (2.0). После этого функция удаляет узел n из дерева случайного поиска. Если окно было на узле, подлежащем удалению, то окно перемещается на предшествующую позицию. Алгоритм реализуется в виде функции _remove, которая удаляет узел n из екущего дерева поиска.

template
void RandomizedSearchTree::_remove(RandomizedNode *n)
{
  n—>_priority = 1.5;
  n->bubbleDown();
  RandomizedNode *p = n->parent();
  if (p->lchild() == n);
    p->_lchild = NULL;
  else
    p->_rchild = NULL;
  if (win == n)
    win = n->prev();
  n->Node::remove();
  delete n;
}

На рис. 3, если его анализировать от (г) до (а), показано удаление элемента 6 из самого правого дерева поиска, для которого приоритет был увеличен до 1.5. Процесс удаления обратен процессу внесения элемента 6 в дерево (а), поскольку правая и левая ротации взаимно обратны.

Для удобства введем вторую компонентную функцию remove, которая, при передаче ей элемента, удаляет его из дерева случайного поиска. Если окно находится на элементе, подлежащем удалению, то окно перемещаетвя в предыдущую позицию, в противном случае окно не перемещается:

template Т RandomizedSearchTree::remove (T val)
{
  Т v = find (val);
  if (v) {
    remove();
    return v;
  }
  return NULL;
}

Ожидаемые характеристики

В дереве случайного поиска размера n средняя глубина узла равна O(log n), Это справедливо независимо от изменений в порядке ввода элементов или положения в дереве, куда эти элементы вводятся или из которого удаляются. Поскольку выполнение каждой из операций ввода и удаления insert, remove и removeMin, а также операций поиска find и findMin занимает время, пропорциональное глубине обрабатываемого элемента, то в среднем затраты времени равны 0(log n).

Почему же средняя длина поиска равна O(log n) ? Чтобы ответить на этот вопрос, найдем ожидаемую глубину элемента в дереве случайного поиска Т. Переименуем элементы в дереве поиска Т по увеличивающемуся приоритету как x₁, x₂,..., x_n. Для определения ожидаемой глубины элемента зд предположим, что дерево Т строилось путем занесения элементов в указанном порядке в первоначально пустое дерево случайного поиска Т'. По мере ввода каждого элемента x_i определим вероятность того, что оно лежит на пути поиска элемента х_k в дереве Т.

Прежде, чем продолжать дальше, обсудим три момента. Во-первых, процесс имеет место в дереве Т, поскольку дерево случайного поиска полностью определяется его элементами и их приоритетами (если даже некоторые значения приоритетов встречаются более одного раза, то элементы все равно могут быть упорядочены таким образом, что дерево Т существует). Во-вторых, вследствие принятого порядка включения элементов не требуется никаких операций ротаций. В-третьих, наш анализ требует, чтобы были внеcены только элементы х₁,..., x_k1, поскольку элемент х_k всегда обязательно содится в конце своего пути поиска и внесение элементов х_k+1,..., x_n не может увеличить глубину элемента x_k, поскольку ни один из этих элементов не может быть предшественником элемента x_k.

Начнем с пустого двоичного дерева Т', представленного единственным аешним узлом. Для внесения элемента х₁ в Т' заменим внешний узел на 1. Поскольку элемент x₁ находится в корне дерева Т', то он будет предшественником любого элемента x_k, так что вероятность того, что элемент x₁ лежит на пути поиска элемента x_k равна единице. Внесем теперь в дерево элемент x₂. Очевидно, что элемент x₂ может заменить любой из двух внешних узлов, которые являются последователями элемента х₁, так что только одна из этих двух позиций приведет к тому, что элемент x₂ будет лежать на пути поиска элемента x_k. Поэтому вероятность того, что элемент x₂ будет лежать на этом пути поиска, равна 1/2. В общем случае, при вводе в дерево Т' элемента x_i, где i изменяется от 1 до k, элемент x_i с равной вероятностью замещает один из i внешних узлов. Поскольку только одна из этих позиций приводит к тому, что элемент х_i будет лежать на пути поиска цемента х_k, то вероятность того, что элемент x_i лежит на пути поиска эле-ента x_k будет равна 1/i. Из этого следует, что ожидаемая глубина элемента равна 1 + СУММА(1/i), i=1..k-1, что соответствует O(log n) для k <= n.