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