链表和数组不一样,它不能直接通过下标来访问对应的元素,只能通过遍历来获取链表的长度以及对应元素,但链表的好处在于,它能够轻松地添加和删除某个元素。
链表结构定义:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
看一道题:
“给定一个头结点为 head
的非空单链表,返回链表的中间结点。
如果有两个中间结点,则返回第二个中间结点。”
例:输入:[1,2,3,4,5]
输出:此列表中的结点 3 (序列化形式:[3,4,5])
返回的结点值为 3 。 (测评系统对该结点序列化表述是 [3,4,5])。
注意,我们返回了一个 ListNode 类型的对象 ans,这样:
ans.val = 3, ans.next.val = 4, ans.next.next.val = 5, 以及 ans.next.next.next = NULL.
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/middle-of-the-linked-list
有两种方法来做这道题,一个是暴力遍历,得到链表的长度,那么中间结点就是长度为一半时的值。代码:
class Solution:
def middleNode(self, head: ListNode) -> ListNode:
# 直接遍历链表,求长度,取中间值
A = [head] # head 直接指向第一个元素,将元素存在列表里
while A[-1].next: # next 是指向下一个元素的指针
A.append(A[-1].next)
return A[len(A) // 2]
还有一个很有趣的方法,叫快慢指针法。它的原理是,先初始化两个指针,都指向第一个元素,然后快指针一次走两步,慢指针一次走一步,当快指针走到链表末尾的时候,慢指针刚好走到链表的一半位置。
代码:
class Solution:
def middleNode(self, head: ListNode) -> ListNode:
# 快慢指针法
slow = fast = head #初始化指针
while fast and fast.next:
slow = slow.next #慢指针走一步
fast = fast.next.next #快指针走两步
return slow
以及另一道基础题:
“给你一个链表,删除链表的倒数第 n
个结点,并且返回链表的头结点。”
例:
输入:head = [1,2,3,4,5], n = 2 输出:[1,2,3,5]
这个我的想法很简单,就是找到倒数第n个结点,并将这个结点的前一个结点的指针修改成指向下下个结点。
代码:
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
a=[head]
while a[-1].next :
a.append(a[-1].next)
b=a[len(a)-n-1]
if len(a)==1:
return None
elif n == len(a):
return head.next
else:
b.next=b.next.next
return head
值得注意的是特殊情况的处理,例如链表里原本只有一个结点,删除后将会是一个空的链表,需返回None(Python里没有NULL...我因为这个之前老是报错),或者是当删除的是头结点,返回的将是头结点的下一个元素。