Python单链表

Python单链表

class Node:
    '''节点 '''
    def __init__(self, elem=None):
        self.elem = elem
        self.next = None

class SingleLinkList:
    '''单链表'''
    def __init__(self, node=None):
        self.__head = node

    def is_empty(self):
        '''链表是否为空'''
        return self.__head == None

    def length(self):
        '''链表长度'''
        cur = self.__head    # cur 游标，用来移动遍历节点
        count = 0            # count 记数
        while cur != None:
            count += 1
            cur = cur.next
        return count

    def travel(self):
        '''遍历整个链表'''
        cur = self.__head
        while cur != None:
            print(cur.elem, end=' ')
            cur = cur.next
        print()

    def add(self, item):
        '''链表头部添加元素，头插法'''
        cur = Node(item)
        cur.next = self.__head
        self.__head = cur

    def append(self, item):
        '''链表尾部添加元素，尾插法'''
        node = Node(item)
        if self.is_empty():
            self.__head = node
        else:
            cur = self.__head
            while cur.next != None:
                cur = cur.next
            cur.next = node

    def insert(self, pos, item):
        '''指定位置添加元素
        :param pos 位置，从 0 开始
        '''
        if pos <= 0:
            self.add(item)
        elif pos > self.length()-1:
            self.append(item)
        else:
            node = Node(item)
            pre = self.__head
            count = 0
            while count < pos-1:
                count += 1
                pre = pre.next
            node.next = pre.next   # 循环退出后，pre指向pos-1位置
            pre.next = node

    def remove(self, item):
        '''删除节点'''
        cur = self.__head
        pre = None
        while cur != None:
            if cur.elem == item:
                if cur == self.__head:      # 判断该节点是否为头结点
                    self.__head = cur.next
                else:
                    pre.next = cur.next
                break
            else:
                pre = cur
                cur = cur.next

    def search(self, item):
        '''查找节点是否存在'''
        cur = self.__head
        while cur != None:
            if cur.elem == item:
                return True
            else:
                cur = cur.next
        return False

1. 创建单链表、判断链表是否为空、查看链表长度

sll = SingleLinkList()    # 创建单链表

sll.is_empty()    # 判断链表是否为空

True

sll.length()      # 链表长度

0

2. 插入链表、遍历链表

• 尾部插入

  # 将数字1 2 3 4 5 6从尾部插入链表
  sll.append(1)
  sll.append(2)
  sll.append(3)
  sll.append(4)
  sll.append(5)
  sll.append(6) 
  sll.travel()     # 遍历链表
  

  1 2 3 4 5 6 
  

• 头部插入

  sll.add(10)      # 将数字10从头部插入链表
  sll.travel()
  

  10 1 2 3 4 5 6 
  

• 指定位置插入

  sll.insert(-1, 20)      # 将数字20在链表位置-1处插入（位置小于等于0时从链表头部插入）
  sll.travel()
  

  20 10 1 2 3 4 5 6
  

  sll.insert(2, 30)      # 将数字30在链表位置2处插入
  sll.travel()
  

  20 10 30 1 2 3 4 5 6 
  

  sll.insert(100, 40)    # 将数字40在链表位置100处插入（位置超出链表长度时从尾部插入）
  sll.travel()
  

  20 10 30 1 2 3 4 5 6 40 
  

3. 删除链表数据

sll.remove(20)         # 将数字20从链表中删除  
sll.travel()

10 30 1 2 3 4 5 6 40

sll.remove(40)        # 将数字40从链表中删除   
sll.travel()

10 30 1 2 3 4 5 6 

sll.remove(3)         # 将数字3从链表中删除 
sll.travel()

10 30 1 2 4 5 6

4. 查看数据是否存在链表

sll.search(30)        # 数字30是否存在于链表中

True

sll.search(40)       # 数字40是否存在于链表中

False