[2] adding ll_find

lukovnikovde/docs/data/DataStructure.py

@@ -1,6 +1,5 @@
 import random as rnd
 #############################################################################################
-head = None
 
 def ll_insert(head, name, phone):
     next_node = {'name': name, 'phone': phone, 'next': None}
@@ -18,10 +17,28 @@ def ll_insert(head, name, phone):
     running['next'] = next_node
     return head
 
-print('======== TESTING LL_INSERT ==========')
-Name = ['Dima', 'Ivan', 'Maxim', 'Alex']
+def ll_find(head, name):
+    running = head
+    flag = True
 
-for _ in range(10):
+    while flag:
+        if running['name'] == name:
+            return running['phone']
+        running = running['next']
+        flag = running is not None
+
+    return None
+
+
+
+def main():
+
+    head = None
+
+    print('======== TESTING LL_INSERT ==========')
+    Name = ['Dima', 'Ivan', 'Maxim', 'Alex']
+
+    for _ in range(10):
         name = Name[rnd.randint(0, 3)]
         phone = str(rnd.randint(0,9)) + str(rnd.randint(0,9)) + str(rnd.randint(0,9)) + '-' + \
                 str(rnd.randint(0,9)) + str(rnd.randint(0,9)) + str(rnd.randint(0,9)) 
@@ -29,5 +46,20 @@ for _ in range(10):
         head = ll_insert(head, name, phone)
         print(head)
         print('-----------------------------------\n')
+    print('======== END TESTING ================')
+
+
+    print('======== TESTING LL_FIND ==========') 
+    Name.append('Masha')
+    for i in range(len(Name)):
+        name = Name[i]
+        print(name, ":", ll_find(head, name))
+    print("======== END TESTING =============")
+
+
+if __name__ == "__main__":
+    main()
+
+
+
 
-print('======== END TESTING ================')