【CV炼丹师勇闯力扣训练营 Day3】

CV炼丹师勇闯力扣训练营

代码随想录算法训练营第三天| 203.移除链表元素、707.设计链表、206.反转链表

---

预备知识：定义链表

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

一、203 移除链表元素

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。
【思路】因为单链表的特殊性，只能指向下一个节点，删除某个节点的方法是找到其前一个结点修改其指针，但是如果删除的是头结点，需要设置一个虚拟头结点再进行删除操作，或者直接将头节点向后移一位。

代码如下（Python）：

（版本一）虚拟头节点法
# 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]:
        # 创建虚拟头部节点以简化删除过程
        dummy_head = ListNode(next = head)
        
        # 遍历列表并删除值为val的节点
        current = dummy_head
        while current.next:
            if current.next.val == val:
                current.next = current.next.next
            else:
                current = current.next
        
        return dummy_head.next

二、707 设计链表

你可以选择使用单链表或者双链表，设计并实现自己的链表。

单链表中的节点应该具备两个属性：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 的节点。
  【注意】遍历链表的时候，没有直接操作头指针，而是通过定义一个cur指针，原因是最终要返回head，直接操作头指针，head已经变了，所以定义零时指针cur，指向头指针

代码如下（Python）：

# 单链表法
class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next

class MyLinkedList:
    def __init__(self):
        self.dummy_head = ListNode()
        self.size = 0

    def get(self, index):
        # 判断n的合法性
        if index < 0 or index >= self.size:
            return -1

        cur = self.dummy_head.next  # 【注意】这里是.next，若不加next，当index为0的时候，返回的是dummy_head的val
        for i in range(index):  # 如果idx=0,则不执行循环
            cur = cur.next

        return cur.val

    def addAtHead(self, val):
        self.dummy_head.next = ListNode(val, self.dummy_head.next)  # 先创建指向self.dummy_head.next的新节点，再更新self.dummy_head.next
        self.size += 1

    def addAtTail(self, val):
        cur = self.dummy_head  # todo: 这里加不加next ?
        # 找尾部节点，即cur.next为空
        while cur.next:
            cur = cur.next
        cur.next = ListNode(val)
        self.size += 1

    def addAtIndex(self, index, val):
        """
        要在第n个节点前插入一个节点，就得知道第n-1个节点
        """
        # 判断n的合法性
        if index < 0 or index > self.size:
            return

        cur = self.dummy_head  # 【注意】这里没有.next，若加next，当index为0的时候，就不是第n个节点前，而是是第n个节点后了
        # 寻找第n-1个节点
        for i in range(index):
            cur = cur.next
        # 第n个节点，cur.next
        cur.next = ListNode(val, cur.next)  # 先创建指向cur.next的节点，再更新cur.next
        self.size += 1

    def deleteAtIndex(self, index: int) -> None:
        """
        要删除第n个节点，就得知道第n-1个节点，操作第n-1个节点的next指针
        第n个节点是cur.next，我们操作cur节点来删掉cur.next节点
        """
        # 判断n的合法性
        if index < 0 or index >= self.size:
            return

        cur = self.dummy_head
        # 寻找第n-1个节点
        for i in range(index):
            cur = cur.next
        cur.next = cur.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)

三、206 反转链表

给你单链表的头节点 head ，请你反转链表，并返回反转后的链表。

首先定义一个cur指针，指向头结点，再定义一个pre指针，初始化为null。
然后就要开始反转了，首先要把 cur->next 节点用tmp指针保存一下，也就是保存一下这个节点。
为什么要保存一下这个节点呢，因为接下来要改变 cur->next 的指向了，将cur->next 指向pre ，此时已经反转了第一个节点了。
接下来，就是循环走如下代码逻辑了，继续移动pre和cur指针。
最后，cur 指针已经指向了null，循环结束，链表也反转完毕了。 此时我们return pre指针就可以了，pre指针就指向了新的头结点。

双指针动画如下所示，纠错应当先移动pre，再移动cur

代码如下（Python）：

（版本一）双指针法
class Solution:
    def reverseList(self, head):
        cur = head
        pre = None
        while cur:
            temp = cur.next  # 保存一下 cur的下一个节点，因为接下来要改变cur->next
            cur.next = pre  # 反转
            # 更新pre、cur指针
            pre = cur  # 【注意】先移动pre再移动cur
            cur = temp
        return pre


（版本二）递归法
class Solution:
    def reverseList(self, head):
        return self.reverse(head, None)  # cur = head, pre = None
    def reverse(self, cur, pre):
        if cur == None:
            return pre  # 当cur为空的时候，递归循环终止，返回头节点
        temp = cur.next  # 保存cur.next
        cur.next = pre  # 反转
        return self.reverse(temp, cur)  # pre = cur, cur = temp