6-链表专题

链表和数组不一样，它不能直接通过下标来访问对应的元素，只能通过遍历来获取链表的长度以及对应元素，但链表的好处在于，它能够轻松地添加和删除某个元素。

链表结构定义：

# 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...我因为这个之前老是报错），或者是当删除的是头结点，返回的将是头结点的下一个元素。