代码随想录算法训练营第三天|链表理论基础、LeetCode203 移除链表元素、LeetCode707 设计链表、LeetCode206 反转链表

代码随想录算法训练营第三天|链表理论基础、LeetCode203 移除链表元素、LeetCode707 设计链表、LeetCode206 反转链表

---

链表理论基础

• 链表在内存中可不是连续分布
• 构造链表：

class ListNode:
    def __init__(self, val, next=None):
        self.val = val
        self.next = next

• 数组和链表的性能分析
  • 数组：
    • 插入/删除时间复杂度O(n)
    • 查询时间复杂度O(1)
    • 适用场景：数据量固定，频繁查询，较少增删
  • 链表：
    • 插入/删除时间复杂度O(1)
    • 查询时间复杂度O(n)
    • 适用场景：数据量不固定，频繁增删，较少查询

---

LeetCode203 移除链表元素

题目链接：https://leetcode.cn/problems/remove-linked-list-elements/description/

• 给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。

---

代码描述：

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def removeElements(self, head: Optional[ListNode], val: int) -> Optional[ListNode]:
        ''' 虚拟头节点法
		时间复杂度：O(n)
		空间复杂度：O(1)
		'''
        dummy_head = ListNode(next = head)		# 创建虚拟头节点，简化删除过程
        current = dummy_head			# 用current存储虚拟头节点，避免因头节点动而无法返回
        while current.next:				# 遍历列表
            if current.next.val == val:
                current.next = current.next.next
            else:
                current = current.next
				
        return dummy_head.next			# 返回虚拟头节点的下一位，也就是真正的头节点

---

LeetCode707 设计链表

题目链接：https://leetcode.cn/problems/design-linked-list/description/

• 你可以选择使用单链表或者双链表，设计并实现自己的链表。
  单链表中的节点应该具备两个属性：val 和 next 。val 是当前节点的值，next 是指向下一个节点的指针/引用。
  如果是双向链表，则还需要属性 prev 以指示链表中的上一个节点。假设链表中的所有节点下标从 0 开始。
  实现 MyLinkedList 类：
  MyLinkedList() 初始化 MyLinkedList 对象。
  int get(int index) 获取链表中下标为 index 的节点的值。如果下标无效，则返回 -1 。
  void addAtHead(int val) 将一个值为 val 的节点插入到链表中第一个元素之前。在插入完成后，新节点会成为链表的第一个节点。
  void addAtTail(int val) 将一个值为 val 的节点追加到链表中作为链表的最后一个元素。
  void addAtIndex(int index, int val) 将一个值为 val 的节点插入到链表中下标为 index 的节点之前。如果 index 等于链表的长度，那么该节点会被追加到链表的末尾。如果 index 比长度更大，该节点将 不会插入 到链表中。
  void deleteAtIndex(int index) 如果下标有效，则删除链表中下标为 index 的节点。

---

代码实现

class LinkNode:
    def __init__(self, val = 0, next = None):
        self.val = val
        self.next = next

class MyLinkedList:
    ''' 单链表法
        '''
    def __init__(self):
        self.dummy_head = LinkNode()
        self.size = 0

    def get(self, index: int) -> int:
        if index < 0 or index >= self.size:
            return -1
        else:
            current = self.dummy_head.next      # 查询节点可以直接定位到该节点位置，所以直接定义current为实际头节点

            for i in range(index):              # 遍历节点到需要查询到节点位置
                current = current.next
            
            return current.val


    def addAtHead(self, val: int) -> None:
        current = LinkNode(val,self.dummy_head.next)
        self.dummy_head.next = current
        # self.dummy_head.next = LinkNode(val,self.dummy_head.next) 可以直接写成这样，记住这个操作
        self.size += 1



    def addAtTail(self, val: int) -> None:
        current = self.dummy_head
        while current.next:
            current = current.next
        current.next = LinkNode(val)
        self.size += 1


    def addAtIndex(self, index: int, val: int) -> None:
        if index < 0 or index > self.size:
            pass
        else:
            current = self.dummy_head       # 插入节点需要先找到前一个结点，所以定义current为虚拟头节点
            for i in range(index):           # 遍历节点到需要插入的节点的前一个位置
                current = current.next
            current.next = LinkNode(val, current.next)
            self.size += 1



    def deleteAtIndex(self, index: int) -> None:
        if index < 0 or index >= self.size:
            pass
        else:
            current = self.dummy_head       # 删除节点需要先找到前一个结点，所以定义current为虚拟头节点
            for i in range(index):           # 遍历节点到需要删除的节点的前一个位置
                current = current.next
            current.next = current.next.next
            self.size -= 1



# Your MyLinkedList object will be instantiated and called as such:
# obj = MyLinkedList()
# param_1 = obj.get(index)
# obj.addAtHead(val)
# obj.addAtTail(val)
# obj.addAtIndex(index,val)
# obj.deleteAtIndex(index)

---

LeetCode206 反转链表

题目链接：https://leetcode.cn/problems/reverse-linked-list/description/

• 题目描述：给你单链表的头节点 head ，请你反转链表，并返回反转后的链表。

---

代码描述

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def reverseList(self, head: Optional[ListNode]) -> Optional[ListNode]:
        ''' 双指针法
            时间复杂度：O(n)
            空间复杂度：O(1)
            '''
        cur = head
        pre = None

        while cur:
            temp = cur.next # 保存cur的下一个节点
            cur.next = pre
            pre = cur
            cur = temp

        return pre

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def reverseList(self, head: Optional[ListNode]) -> Optional[ListNode]:
        ''' 递归法
            时间复杂度：O(n)
            空间复杂度：O(n)
            '''
        return self.reverse(head, None)

    def reverse(self, cur:ListNode, pre:ListNode) -> ListNode:
        if cur == None:
            return pre
        temp = cur.next
        cur.next = pre
        return self.reverse(temp, cur)