forked from avsingh999/Python-for-beginner
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathLinkedList.py
57 lines (46 loc) · 1.37 KB
/
LinkedList.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
class LinkedList:
def __init__(self, data=None):
self.data = data
self.next = None
def prepend(self, element):
newNode = LinkedList(element)
newNode.next = self
return newNode
def append(self, element):
if self.next is None:
self.next = LinkedList(element)
else:
self.next.append(element)
return self
def head(self):
return self.data
def tail(self):
return self.next
def contains(self, element):
if self.data == element:
return True
elif self.next is None:
return False
else:
return self.next.contains(element)
def remove(self, element):
if self.data == element:
return self.next
elif self.data is None or self.next is None:
return self
else:
self.next = self.next.remove(element)
return self
def foreach(self, f):
if self.data is not None:
f(self.data)
if self.next is not None:
self.next.foreach(f)
def map(self, f):
if self.data is not None:
ll = LinkedList(f(self.data))
if self.next is not None:
ll.next = self.next.map(f)
return ll
def __str__(self):
return str(self.data) + ', ' + str(self.next)