From 2c8b9d01df7e1aea9fb7ab9623060f7b59b4016d Mon Sep 17 00:00:00 2001 From: Adam Blank <blank@cs.washington.edu> Date: Mon, 16 Jan 2017 09:02:19 -0800 Subject: [PATCH] Update BinarySearchTree.java --- src/cse332/datastructures/trees/BinarySearchTree.java | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/cse332/datastructures/trees/BinarySearchTree.java b/src/cse332/datastructures/trees/BinarySearchTree.java index e675f8e..92fecb7 100644 --- a/src/cse332/datastructures/trees/BinarySearchTree.java +++ b/src/cse332/datastructures/trees/BinarySearchTree.java @@ -49,7 +49,7 @@ public class BinarySearchTree<K extends Comparable<K>, V> } } - private BSTNode find(K key, V value) { + protected BSTNode find(K key, V value) { BSTNode prev = null; BSTNode current = this.root; @@ -89,6 +89,9 @@ public class BinarySearchTree<K extends Comparable<K>, V> @Override public V find(K key) { + if (key == null) { + throw new IllegalArgumentException(); + } BSTNode result = find(key, null); if (result == null) { return null; @@ -98,6 +101,9 @@ public class BinarySearchTree<K extends Comparable<K>, V> @Override public V insert(K key, V value) { + if (key == null || value == null) { + throw new IllegalArgumentException(); + } BSTNode current = find(key, value); V oldValue = current.value; current.value = value; -- GitLab