92. 反转链表 II
题目
反转从位置 m 到 n 的链表。请使用一趟扫描完成反转。
说明:
1 ≤ m ≤ n ≤ 链表长度。
示例:
输入: 1->2->3->4->5->NULL, m = 2, n = 4
输出: 1->4->3->2->5->NULL
解题思路
第一反应可使用逐个插入法完成,但空间复杂度与实践复杂度都较高。所以主要考虑使用指针方法完成。LeetCode 206.反转链表 是对整个链表的反转,比这个稍微简单,已给出具体解法,本题解法与其中的常规解法类似。
本题解法(命名有点草率):
- 使用指针 this_code 与 last_node 指代当前处理的元素与上一个元素,并用 i 记录元素编号,next_node 指代下一个元素
- 若 i == m,表示开始需要反转,此时,last_node 即为第一个正序列表中最后一个元素,记为 pre;this_node 即为需要反转的链表中第一个元素,记为 fst,同时也是反转链表后的最后一个元素
- 若 i - 1 >= m 且 i <=n 表示当前元素需要反转,此时将 this_node 的 next 指向 last_node
- 若 i == n,表示反转已经结束。此时,需要将已经反转的链表与原来的元素进行连接,即 pre.next = this_node,fst.next = next_node
代码
class Solution:
def reverseBetween(self, head, m, n):
"""
:type head: ListNode
:type m: int
:type n: int
:rtype: ListNode
"""
# 创建一个空的头指针,指向 head
head_code = ListNode(-1)
head_code.next = head
last_node = head_code
this_node = head_code.next
i = 1
while this_node: # 获取当前指针
next_node = this_node.next # 记录下一个node
if i == m: # 反转开始
# 记录上一个结点与当前结点
pre = last_node
fst = this_node
if i - 1 >= m and i <=n: # 当前元素在需反转的链表内
this_node.next = last_node # 修改指针
if i == n: # 反转结束
# 修改指针
pre.next = this_node
fst.next = next_node
# 往后遍历
last_node, this_node = this_node, next_node
i += 1
return head_code.next
结果
速度还是可以的