牛客网算法入门Python版 | 反转链表 | 双指针法

一、题目描述

给定一个单链表的头结点pHead(该头节点是有值的，比如在下图，它的val是1)，长度为n，反转该链表后，返回新链表的表头。

数据范围： 0≤𝑛≤10000≤n≤1000

要求：空间复杂度 O(1) ，时间复杂度 O(n) 。

如当输入链表{1,2,3}时，

经反转后，原链表变为{3,2,1}，所以对应的输出为{3,2,1}。

以上转换过程如下图所示：

示例1

输入：{1,2,3}

返回值：{3,2,1}

示例2

输入：{}

返回值：{}

说明：空链表则输出空

二、算法思想

算法思想：双指针迭代

（1）定义两个指针 pre 和 cur 用来反转，再定义一个指针 temp 用于存储后面结点防止断链；
（2）每次让 pre 的 next 指向 cur ，实现一次局部反转；
（3）局部反转完成之后， pre 和 cur 同时往下移动一个位置；
（4）循环上述过程，直至 pre 到达链表尾。

三、代码

# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None
#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param head ListNode类 
# @return ListNode类
#
class Solution:
    def ReverseList(self , head: ListNode) -> ListNode:
        #处理空链表
        if not head:
            return None
        #设立这两个“指针”来实现反转
        cur = head #设立一个当前节点
        pre = None # 设立当前节点的前一个节点，初始值为空，最终的结果是反转后的头结点
        while cur:
            temp = cur.next #找到后续一个节点，防止断链后找不到
            cur.next = pre #解题核心：反转链表
            pre = cur #向后移动，以便迭代
            cur = temp #向后移动，以便迭代
        return pre #返回头结点

四、学习笔记

为什么 return pre？

对于单链表来说，返回链表头结点能够返回整个链表。