Merge pull request joeyajames#10 from tomas2310328/master

joeyajames · web-flow · commit 85cc70e62368 · 2019-04-17T09:24:12.000-07:00
Adding Keys function to return all values in hashMap.py
diff --git a/HashMap.py b/HashMap.py
@@ -1,55 +1,62 @@
 # Hash Map
 
 class HashMap:
-	def __init__(self):
-		self.size = 6
-		self.map = [None] * self.size
+        def __init__(self):
+                self.size = 6
+                self.map = [None] * self.size
 		
-	def _get_hash(self, key):
-		hash = 0
-		for char in str(key):
-			hash += ord(char)
-		return hash % self.size
+        def _get_hash(self, key):
+                hash = 0
+                for char in str(key):
+                        hash += ord(char)
+                return hash % self.size
 		
-	def add(self, key, value):
-		key_hash = self._get_hash(key)
-		key_value = [key, value]
+        def add(self, key, value):
+                key_hash = self._get_hash(key)
+                key_value = [key, value]
 		
-		if self.map[key_hash] is None:
-			self.map[key_hash] = list([key_value])
-			return True
-		else:
-			for pair in self.map[key_hash]:
-				if pair[0] == key:
-					pair[1] = value
-					return True
-			self.map[key_hash].append(key_value)
-			return True
+                if self.map[key_hash] is None:
+                        self.map[key_hash] = list([key_value])
+                        return True
+                else:
+                        for pair in self.map[key_hash]:
+                                if pair[0] == key:
+                                        pair[1] = value
+                                        return True
+                        self.map[key_hash].append(key_value)
+                        return True
 			
-	def get(self, key):
-		key_hash = self._get_hash(key)
-		if self.map[key_hash] is not None:
-			for pair in self.map[key_hash]:
-				if pair[0] == key:
-					return pair[1]
-		return None
+        def get(self, key):
+                key_hash = self._get_hash(key)
+                if self.map[key_hash] is not None:
+                        for pair in self.map[key_hash]:
+                                if pair[0] == key:
+                                        return pair[1]
+                return None
 			
-	def delete(self, key):
-		key_hash = self._get_hash(key)
+        def delete(self, key):
+                key_hash = self._get_hash(key)
 		
-		if self.map[key_hash] is None:
-			return False
-		for i in range (0, len(self.map[key_hash])):
-			if self.map[key_hash][i][0] == key:
-				self.map[key_hash].pop(i)
-				return True
-		return False
+                if self.map[key_hash] is None:
+                        return False
+                for i in range (0, len(self.map[key_hash])):
+                        if self.map[key_hash][i][0] == key:
+                                self.map[key_hash].pop(i)
+                                return True
+                return False
+	
+        def keys(self):
+                arr = []
+                for i in range(0, len(self.map)):
+                        if self.map[i]:
+                                arr.append(self.map[i][0])
+                return arr
 			
-	def print(self):
-		print('---PHONEBOOK----')
-		for item in self.map:
-			if item is not None:
-				print(str(item))
+        def print(self):
+                print('---PHONEBOOK----')
+                for item in self.map:
+                        if item is not None:
+                                print(str(item))
 			
 h = HashMap()
 h.add('Bob', '567-8888')
@@ -64,4 +71,4 @@ def print(self):
 h.delete('Bob')
 h.print()
 print('Ming: ' + h.get('Ming'))
-		
+print(h.keys())
