// Алгоритм рекурсивного добавления нового узла в дерево. private void AddTo(AvlTreeNode <T> node, T value) { if (value.CompareTo(node.Value) < 0) //Добавление нового узла в дерево. Если значение добавлемого узла меньше чем значение текущего узла. { if (node.Left == null) //Создание левого узла, если его нет. { node.Left = new AvlTreeNode <T>(value, node, this); } else { // Переходим к следующему левому узлу AddTo(node.Left, value); } } else // Если добавлемое значение больше или равно текущему значению. { if (node.Right == null) //Создание правого узла, если его нет. { node.Right = new AvlTreeNode <T>(value, node, this); } else { // Переход к следующему правому узлу. AddTo(node.Right, value); } } node.Balance(); }
// Метод удаляет элемент по значению public bool Remove(T value) { AvlTreeNode <T> current; current = Find(value); // находим узел с удаляемым значением if (current == null) // узел не найден { return(false); } AvlTreeNode <T> treeToBalance = current.Parent; // баланс дерева относительно узла родителя Count--; // уменьшение колиества узлов //Если удаляемый узел не имеет правого потомка if (current.Right == null) { if (current.Parent == null) // удаляемый узел является корнем { Head = current.Left; // на место корня перемещаем левого потомка if (Head != null) { Head.Parent = null; // убераем ссылку на родителя } } else // удаляемый узел не является корнем { int result = current.Parent.CompareTo(current.Value); if (result > 0) { // Если значение родительского узла больше значения удаляемого, сделать левого потомка удаляемого узла, левым потомком родителя. current.Parent.Left = current.Left; } else if (result < 0) { // Если значение родительского узла меньше чем удаляемого, // сделать левого потомка удаляемого узла - правым потомком // родительского узла.current.Parent.Right = current.Left; } } } // Если правый потомок удаляемого узла не имеет левого потомка, тогда правый потомок удаляемого узла становится потомком родительского узла. else if (current.Right.Left == null) // если у правого потомка нет левого потомка { current.Right.Left = current.Left; if (current.Parent == null) // текущий элемент является корнем { Head = current.Right; if (Head != null) { Head.Parent = null; } } else { int result = current.Parent.CompareTo(current.Value); if (result > 0 ) // Если значение узла родителя больше чем значение удаляемого узла, сделать правого потомка удаляемого узла, левым потомком его родителя. { current.Parent.Left = current.Right; } else if (result < 0 ) // Если значение родительского узла меньше значения удаляемого,сделать правого потомка удаляемого узла - правым потомком родителя. { current.Parent.Right = current.Right; } } } // Если правый потомок удаляемого узла имеет левого потомка,заместить удаляемый узел, крайним левым потомком правого потомка. else { // Нахожление крайнего левого узла для правого потомка удаляемого узла. AvlTreeNode <T> leftmost = current.Right.Left; while (leftmost.Left != null) { leftmost = leftmost.Left; } // Родительское правое поддерево становится родительским левым поддеревом. leftmost.Parent.Left = leftmost.Right; // Присвоить крайнему левому узлу, ссылки на правого и левого потомка удаляемого узла. leftmost.Left = current.Left; leftmost.Right = current.Right; if (current.Parent == null) { Head = leftmost; if (Head != null) { Head.Parent = null; } } else { int result = current.Parent.CompareTo(current.Value); if (result > 0) { // Если значение родительского узла больше значения удаляемого,сделать крайнего левого потомка левым потомком родителя удаляемого узла. current.Parent.Left = leftmost; } else if (result < 0) { // Если значение родительского узла, меньше чем значение удаляемого,сделать крайнего левого потомка, правым потомком родителя удаляемого узла. current.Parent.Right = leftmost; } } } if (treeToBalance != null) { treeToBalance.Balance(); } else { Head?.Balance(); } return(true); }