Привёл add в MyTreeSet в соответствие с интерфейсом Set
This commit is contained in:
@@ -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));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user