跟随代码随想录和编程文青李狗蛋学习
203. 移除链表元素https://leetcode.cn/problems/remove-linked-list-elements/给你一个链表的头节点
head
和一个整数 val
,请你删除链表中所有满足 Node.val == val
的节点,并返回 新的头节点 。
示例 1:
输入:head = [1,2,6,3,4,5,6], val = 6
输出:[1,2,3,4,5]
示例 2:
输入:head = [], val = 1
输出:[]
示例 3:
输入:head = [7,7,7,7], val = 7
输出:[]
思路:建立一个虚拟头结点,使用双指针前进,当快的指针指向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]:
dummy = ListNode(0)
dummy.next = head
prev = dummy
while head is not None:
if head.val == val:
prev.next = head.next
else:
prev = prev.next
head = head.next
return dummy.next
707. 设计链表https://leetcode.cn/problems/design-linked-list/
设计链表的实现。您可以选择使用单链表或双链表。单链表中的节点应该具有两个属性:val 和 next。val 是当前节点的值,next 是指向下一个节点的指针/引用。如果要使用双向链表,则还需要一个属性 prev 以指示链表中的上一个节点。假设链表中的所有节点都是 0-index 的。
在链表类中实现这些功能:
get(index):获取链表中第 index 个节点的值。如果索引无效,则返回-1。
addAtHead(val):在链表的第一个元素之前添加一个值为 val 的节点。插入后,新节点将成为链表的第一个节点。
addAtTail(val):将值为 val 的节点追加到链表的最后一个元素。
addAtIndex(index,val):在链表中的第 index 个节点之前添加值为 val 的节点。如果 index 等于链表的长度,则该节点将附加到链表的末尾。如果 index 大于链表长度,则不会插入节点。如果index小于0,则在头部插入节点。
deleteAtIndex(index):如果索引 index 有效,则删除链表中的第 index 个节点。
示例:
MyLinkedList linkedList = new MyLinkedList();
linkedList.addAtHead(1);
linkedList.addAtTail(3);
linkedList.addAtIndex(1,2); //链表变为1-> 2-> 3
linkedList.get(1); //返回2
linkedList.deleteAtIndex(1); //现在链表是1-> 3
linkedList.get(1); //返回3
单链表:
class Node:
def __init__(self, data):
self.val = data
self.next = None
class MyLinkedList:
def __init__(self):
# 头节点
self.head = Node(0)
# 保存链表长度
self.length = 0
def get(self, index: int) -> int:
# 判断输入的索引是否有效
if index < 0 or index >= self.length:
return -1
p = self.head
# range 是左闭右开区间,相当于[0,index+1)
for _ in range(index + 1):
p = p.next
return p.val
def addAtHead(self, val: int) -> None:
return self.addAtIndex(0, val)
def addAtTail(self, val: int) -> None:
return self.addAtIndex(self.length, val)
def addAtIndex(self, index: int, val: int) -> None:
# 特殊情况 index < 0 和 index = 0,则说明是头插法
if index < 0:
index = 0
if index > self.length:
return
p = self.head
add_node = Node(val)
# 找到第 index 个节点的前驱节点
for _ in range(index):
p = p.next
add_node.next = p.next
p.next = add_node
# 插入节点,链表长度 +1
self.length += 1
def deleteAtIndex(self, index: int) -> None:
# 判断输入的索引是否有效
if index < 0 or index >=self.length:
return
p = self.head
# 找到要删除节点的前驱节点和要删除的节点
for _ in range(index + 1):
# 前驱节点
pre = p
# 要删除节点
p = p.next
pre.next = p.next
# 删除节点,链表长度 -1
self.length -= 1
206. 反转链表https://leetcode.cn/problems/reverse-linked-list/给你单链表的头节点
head
,请你反转链表,并返回反转后的链表。
示例 1:
输入:head = [1,2,3,4,5]
输出:[5,4,3,2,1]
示例 2:
输入:head = [1,2]
输出:[2,1]
示例 3:
输入:head = []
输出:[]
思路:使用temp变量来保存当前节点的next指针。一定要先保存。
#双指针
# 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: ListNode) -> ListNode:
if not head or head.next == None:
return head
cur = head
pre = None
while cur:
temp = cur.next # 保存一下 cur的下一个节点,因为接下来要改变cur->next
cur.next = pre #反转
#更新pre、cur指针
pre = cur
cur = temp
return pre
跟随代码随想录和编程问清李狗蛋
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/remove-linked-list-elements
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。