首页技术文章正文

如何区分python中的链表和数组?

更新时间:2022-08-31 来源:黑马程序员 浏览量:

链表是节点的集合。第一个节点(Node)一般被称为Head。最后一个节点的Next属性必须指向 None ,表明是链表的结尾。

1661916568939_链表.png

在大多数编程语言中,链表和数组在内存中的存储方式存在明显差异。数组要求内存空间是连续的,链表可以不连续。

然而,在 Python 中,list是动态数组。所以在Python中列表和链表的内存使用非常相似。

链表和数组在以下的操作中也有本质区别:

1.插入元素:数组中插入元素时,插入位置之后的所有元素都需要往后移动一位,所以数组中插入元素最坏时间复杂度是 O(n)链表可以达到 O(1) 的时间复杂度。

2. 删除元素:数组需要将删除位置之后的元素全部往前移动一位,最坏时间复杂度是 O(n),链表可以达到 O(1) 的时间复杂度。

3.随机访问元素:数组可以通过下标直接访问元素,时间复杂度为O(1)。链表需要从头结点开始遍历,时间复杂度为O(n)。

4.获取长度: 数组获取长度的时间复杂度为O(1),链表获取长度也只能从头开始遍历,时间复杂度为O(n)。
实现链表

class LinkedList:
   def __init__(self):
       self.head = None

在LinkedList中,需要存储的唯一信息是链表的开始位置(链表的头部)。接下来,创建另一个类Node来表示链表的每个节点:

class Node:
    def __init__(self, data):
        self.data = data
        self.next = None

我们可以给刚创建的两个类添加 __repr__ 方法, 在创建实例的时候输出更多有用的信息:

class LinkedList:
    def __init__(self):
        self.head = None
def __repr__(self):
    node = self.head
    nodes = []
    while node is not None:
        nodes.append(node.data)
        node = node.next
    nodes.append("None")
    return " -> ".join(nodes)

我们可以给刚创建的两个类添加 __repr__ 方法, 在创建实例的时候输出更多有用的信息

class Node:
    def __init__(self, data):
       self.data = data
       self.next = None
        
    def __repr__(self):
            return self.data

创建测试类, 测试上面的代码

from LinkedList import LinkedList
from Node import Node
if __name__ == '__main__':
    llist = LinkedList()
    print(llist)
    first_node = Node('a')
    llist.head = first_node
    print(llist)
    second_node = Node('b')
    third_node = Node('c')
    first_node.next = second_node
    second_node.next = third_node
    print(llist)

修改__init__ 方法,可以传入列表快速创建LinkedList

def __init__(self, nodes=None):
    self.head = None
    if nodes is not None:
        node = Node(data=nodes.pop(0))
    self.head = node
    for elem in nodes:
        node.next = Node(data=elem)
        node = node.next

创建__iter__ 方法,遍历链表

def __iter__(self):
    node = self.head
    while node is not None:
        yield node
        node = node.next
from LinkedList import LinkedList
if __name__ == '__main__':
    llist = LinkedList(['a','b','c','d','e'])
    print(llist)
for node in llist:

    print(node)

实现链表,添加节点

在头部添加Node:在链表的开头添加一个Node,不必遍历链表,只需将新的Node的next属性指向 self.head ,并将新的node设置为新的 self.head

def add_first(self, node):
    node.next = self.head
    self.head = node
from LinkedList import LinkedList
from Node import Node
if __name__ == '__main__':
    llist = LinkedList()
    print(llist)
    llist.add_first(Node('b'))
    print(llist)
    llist.add_first(Node('a'))
    print(llist)

在尾部添加Node:必须遍历链表,与list不同,list可以直接获取长度, 链表只有从第一个Node,不断的去获取下一个Node 才能知道链表的尾部。

def add_last(self, node):
    if self.head is None:
        self.head = node
        return
    for current_node in self:
        pass
    current_node.next = node
from LinkedList import LinkedList
from Node import Node
if __name__ == '__main__':
    llist = LinkedList(['a','b','c','d'])
    print(llist)

    llist.add_last(Node('e'))
    print(llist)

    
    llist.add_last(Node('f'))
    print(llist)

在指定元素后添加Node:遍历链表找到目标Node, 把目标Node的下一个元素, 赋值给要添加Node的next属性, 然后修改目标Node的next属性, 指向新添加的Node, 当链表为空以及目标元素不存在时抛出异常。

def add_after(self, target_node_data, new_node):
   if self.head is None:
        raise Exception("List is empty")
for node in self:
   if node.data == target_node_data:
     new_node.next = node.next
     node.next = new_node
     return
     
     raise Exception("Node with data '%s' not found" % target_node_data)

在指定元素后添加Node:遍历链表找到目标Node, 把目标Node的下一个元素, 赋值给要添加Node的next属性, 然后修改目标Node的next属性, 指向新添加的Node, 当链表为空以及目标元素不存在时抛出异常。

from LinkedList import LinkedList
from Node import Node
if __name__ == '__main__':

    llist = LinkedList()
    llist.add_after("a", Node("b"))
    llist = LinkedList(['a','b','c','d'])
    print(llist)
    llist.add_after("c", Node("cc"))
    print(llist)
    llist.add_after("f", Node("g"))

在指定元素前添加Node:遍历链表找到目标Node,还需要记录当前节点的前一个节点。

def add_before(self, target_node_data, new_node):
    if self.head is None:
        raise Exception("List is empty")

    if self.head.data == target_node_data:
        return self.add_first(new_node)
    prev_node = self.head
    for node in self:
        if node.data == target_node_data:
        prev_node.next = new_node
        new_node.next = node
        return
    prev_node = node

    raise Exception("Node with data '%s' not found" % target_node_data)
from LinkedList import LinkedList
from Node import Node
if __name__ == '__main__’:
    llist = LinkedList()
    llist.add_before("a", Node("b"))
    llist = LinkedList(["b", "c"])
    print(llist)
    llist.add_before('b',Node('a'))
    print(llist)
    llist.add_before("b", Node("aa"))
    llist.add_before("c", Node("bb"))
    print(llist)
    llist.add_before("n", Node("m"))

删除Node:遍历链表找到目标Node,将目标Node的前一个Node的next属性,指向目标Node的next节点。

def remove_node(self, target_node_data):
    if self.head is None:
        raise Exception("List is empty")
    if self.head.data == target_node_data:
        self.head = self.head.next
        return

    previous_node = self.head
    for node in self:
        if node.data == target_node_data:
        previous_node.next = node.next
        return
    previous_node = node
raise Exception("Node with data '%s' not found" % target_node_data)
from LinkedList import LinkedList
from Node import Node
if __name__ == '__main__’:
    llist = LinkedList(["a", "b", "c", "d", "e"])
    print(llist)
    llist.remove_node("a")
    print(llist)
    llist.remove_node('e')
    print(llist)


分享到:
在线咨询 我要报名
和我们在线交谈!