单链表及其基本操作

  链表是最基本的数据结构,很多高级的数据结构都离不开链表的身影,本文从最基本的链表实现开始,逐步实现对单链表的判空、求长度、遍历、添加、删除等操作。

1.定义链表节点

  链表是由一个个数据节点连接而成,首先需要构建数据节点,包含数据元素elem和指向下一节点的next。

1 class Node(object): 
2     """首先定义节点类"""
3     def __init__(self, element):  # 构建数据节点时,需要传入数据元素element
4         self.elem = element
5         self.next = None  # 由于一开始next节点不知道指向什么地方,所以就设置为空

2.初始化单链表

  初始化链表时,需要一个对象属性指向头结点。

1 class SingleLinkList(object):
2     """定义单链表类"""
3     def __init__(self, node = None):  # 构建链表时,需要传入数据节点node,node默认为None
4         self. __head = node  # head是私有属性

3.单链表基本操作

3.1.链表判空

  空链表的头一定是指向None的。

1     def is_empty(self):
2         """链表是否为空"""
3         return self.__head is None

3.2.求链表长度

  求链表长度就是统计链表节点的个数,可以使用一个游标cur遍历节点,使用一个计数器count记录节点个数。

1     def length(self):
2         """链表长度"""
3         cur = self.__head
4         count = 0
5         while cur is not None:
6             count += 1
7             cur = cur.next
8         return count

3.3.遍历链表

1     def travel(self):
2         """遍历链表"""
3         cur = self.__head
4         while cur is not None:
5             print(cur.elem, end=" ")  # end=" "关闭换行功能,print的原型默认是以end="
"结尾.
6             cur = cur.next
7         print(end="
")

3.4.往链表头部添加元素

  往链表中添加元素就相当远添加数据节点,需要传入节点的数据。

  Tips_1:增删链表时,A=B可以被理解为,使A指向B;

  Tips_2:增删链表时,需要先保证不破坏原有链表的链接,以保证操作的顺序正确,如需要先node.next = self.__head,再self.__head = node。

1     def add(self, item):
2         """在链表头部添加元素"""
3         node = Node(item)
4         node.next = self.__head
5         self.__head = node

3.5.往链表尾部添加元素

 1     def append(self, item):
 2         """链表尾部添加元素"""
 3         node = Node(item)
 4         if self.is_empty():
 5             self.__head = node
 6         else:
 7             cur = self.__head
 8             while cur.next is not None:
 9                 cur = cur.next
10             cur.next = node

3.6.往链表指定位置添加元素

  需要考虑添加位置的边界条件:如果添加的位置是链表头部,可以直接调用add()函数。如果添加的位置是链表尾部,可以直接调用append()函数。

 1     def insert(self, pos, item):
 2         """在指定位置添加元素"""
 3         if pos <= 0:
 4             self.add(item)
 5         elif pos > self.length()-1:
 6             self.append(item)
 7         else:
 8             node = Node(item)
 9             prev = self.__head
10             count = 0
11             while count < (pos-1):
12                 count += 1
13                 cur = cur.next
14             node.next = cur.next
15             cur.next = node

3.7.删除节点

  链表被打断后,需要pre记录断点前的链表部分。

 1     def remove(self, item):
 2         """删除节点"""
 3         cur = self.__head
 4         pre = None  
 5         while cur is not None:
 6             if cur.elem == item:
 7                 if cur == self.__head:  # 头节点特殊处理
 8                     self.__head = cur.next
 9                 else:
10                     pre.next = cur.next
11                 break
12             else:
13                 pre = cur
14                 cur = cur.next

3.8.查找节点

1     def search(self, item):
2         """查找节点是否存在"""
3         cur = self.__head
4         while cur is not None:
5             if cur.elem == item:
6                 return True
7             else:
8                 cur = cur.next
9         return False

3.9.测试与结果

 1 # 测试
 2 if __name__ == "__main__":
 3     li = SingleLinkList()
 4     print(li.is_empty())       # True
 5     print(li.length())         # 0
 6 
 7     li.append(1)
 8     print(li.is_empty())       # False
 9     print(li.length())         # 1
10 
11     li.append(2)
12     li.add(8)
13     li.append(3)
14     li.append(4)
15     li.append(5)
16     li.append(6)
17     li.insert(-1, 9)  
18     li.insert(2, 100)  
19     li.insert(10, 200) 
20     li.travel()                # 9 8 100 1 2 3 4 5 6 200
21 
22     li.remove(100)
23     li.travel()                # 9 8 1 2 3 4 5 6 200
24     li.remove(9)  
25     li.travel()                # 8 1 2 3 4 5 6 200 
26     li.remove(200)  
27     li.travel()                # 8 1 2 3 4 5 6
原文地址:https://www.cnblogs.com/kongzimengzixiaozhuzi/p/12951670.html