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