Уже запуталась немного как восстанавливать возможность иметь детей. Сделала так: если больше не осталось вершин,способных иметь детей,то в дереве нахожу все бездетные вершины и добавляю их в главную очередь (leaves).
public boolean add(String s) {
Entry<String> entry = new Entry<>(s);
if (currentNode==null){ //восстановление потомков, если currentNode == null, значит очередь leaves пуста, надо ее восстановить и
recoverLeaves(root);
currentNode = leaves.poll();
}
if (!currentNode.isAvailableToAddChildren()) {
currentNode = leaves.poll();
}
addChild(entry,currentNode);
size++;
return true;
}private void recoverLeaves(Entry<String> rootNode){
Queue<Entry<String>> queue = new LinkedList<>();
Entry<String> elem = rootNode;
do{
if (queue.size()!=0) elem = queue.poll();
if (elem.isChildFree()) {
elem.availableToAddLeftChildren = true;
elem.availableToAddRightChildren = true;
leaves.add(elem);
} else {
if (elem.leftChild!=null) queue.add(elem.leftChild);
if (elem.rightChild!=null) queue.add(elem.rightChild);
};
}while (queue.size()!=0);
}Добавлять в дерево элементы мы можем, теперь займись удалением:
необходимо реализовать метод remove(Object o) для удаления элемента дерева, имя которого было получено в качестве параметра.
Если переданный объект не является строкой, метод должен бросить UnsupportedOperationException.
Если в дереве присутствует несколько элементов с переданным именем - можешь удалить только первый найденный.
Не забывай сверять поведение своего дерева с картинкой:
Что будет, если удалить из дерева элементы "3", "4", "5" и "6", а затем попытаться добавить новый елемент?
В таком случае элементы "1" и "2" должны восстановить возможность иметь потомков (возможно, придется внести изменения в метод add()).Требования:
- После удаления последнего добавленного элемента из дерева с помощью метода remove, метод size должен возвращать N-1.
- После удаления второго элемента добавленного в дерево, метод size должен возвращать N/2 + 1 (для случаев где N > 2 и является степенью двойки), N - размер дерева до удаления.
- Если переданный объект не является строкой, метод remove() должен бросить UnsupportedOperationException.
- Если ни один элемент не способен иметь потомков, необходимо восстановить такую возможность.
package com.javarush.task.task20.task2028;
import java.util.List;
public class Solution {
public static void main(String[] args) {
List<String> list = new CustomTree();
for (int i = 1; i < 16; i++) {
list.add(String.valueOf(i));
}
// System.out.println("The list size is " + list.size());
// System.out.println("The expected parent is 3. The actual parent is " + ((CustomTree) list).getParent("8"));
//System.out.println("The expected parent is null. The actual parent is " + ((CustomTree) list).getParent("20"));
// list.remove("6");
list.remove("3");
System.out.println("The expected parent is null. The actual parent is " + ((CustomTree) list).getParent("8"));
list.add("16");
System.out.println("The expected parent is 9. The actual parent is " + ((CustomTree) list).getParent("16"));
list.remove("4");
list.remove("5");
list.remove("6");
// list.remove("4");
// list.remove("5");
// list.remove("3");
System.out.println("Expected: true. Actual: " + list.add("20"));
System.out.println("The expected parent is 1. The actual parent is " + ((CustomTree) list).getParent("20"));
}
}