Updated

Author: 0xShun

DSA/linked_list.py

@@ -1,27 +1,22 @@
 class Node:
     def __init__(self, data):
         self.data = data
-        self.next = None  # make None as the default value for next.
+        self.next = None
+        
+def insertBeginning(head, data):
+    new_node = Node(data)
+    new_node.next = head
+    return new_node  
+    
+def main():
+    head = None 
+    head = insertBeginning(head, "Shawn")
+    head = insertBeginning(head, "Sean")
+    head = insertBeginning(head, "John")
+    head = insertBeginning(head, "Zyra")
 
-def count_nodes(head):
-    # assuming that head != None
-    count = 1
-    current = head
-    while current.next is not None:
-        current = current.next
-        count += 1
-    return count
+    print(head.next.next.data)
 
-nodeA = Node(6)
-nodeB = Node(3)
-nodeC = Node(4)
-nodeD = Node(2)
-nodeE = Node(1)
 
-nodeA.next = nodeB
-nodeB.next = nodeC
-nodeC.next = nodeD
-nodeD.next = nodeE
-
-print("This linked list's length is: (should print 5)")
-print(count_nodes(nodeA))
+main()
+