Привёл 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 ) public static void main( String[] args )
{ {
MyTreeSet<Integer> tree = new MyTreeSet<>(); MyTreeSet<Integer> tree = new MyTreeSet<>();
tree.add(10); System.out.println(tree.add(10));
tree.add(20); System.out.println(tree.add(20));
tree.add(5); System.out.println(tree.add(10));
} }
} }

View File

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