Привёл add в MyTreeSet в соответствие с интерфейсом Set

This commit is contained in:
2025-02-03 17:27:13 +03:00
parent 4bb503b70f
commit a87c13dc42
2 changed files with 8 additions and 6 deletions

View File

@@ -5,8 +5,8 @@ public class App
public static void main( String[] args )
{
MyTreeSet<Integer> tree = new MyTreeSet<>();
tree.add(10);
tree.add(20);
tree.add(5);
System.out.println(tree.add(10));
System.out.println(tree.add(20));
System.out.println(tree.add(10));
}
}

View File

@@ -19,10 +19,10 @@ public class MyTreeSet<E extends Comparable<E>> {
root = null;
}
public void add(E element) {
public boolean add(E element) {
if (root == null) {
root = new Node(element);
return;
return true;
}
Node currentNode = root;
@@ -31,7 +31,7 @@ public class MyTreeSet<E extends Comparable<E>> {
while (currentNode != null) {
int cmp = element.compareTo(currentNode.value);
if (cmp == 0) {
return;
return false;
}
parentNode = currentNode;
currentNode = cmp < 0 ? currentNode.left : currentNode.right;
@@ -44,5 +44,7 @@ public class MyTreeSet<E extends Comparable<E>> {
} else {
parentNode.right = newNode;
}
return true;
}
}