更多题解,请关注公众号:程序员学长,让你进大厂不走弯路
链表内指定区间反转
问题描述
给你单链表的头指针 head 和两个整数 left 和 right ,其中 left <= right 。请你反转从位置 left 到位置 right 的链表节点,返回反转后的链表 。
示例:
输入:head = [1,2,3,4,5], left = 2,right = 4
输出:[1,4,3,2,5]
分析问题
对于链表相关的题目,我们一定要先想清楚思路,搞懂指针移动的先后顺序。
对于这道题,我们可以采用头插法来求解。在链表反转的区间内,当我们每次遍历到一个节点时,就让该节点插入到反转部分的起始位置。如下图所示:
具体来说,我们定义三个指针pre、cur、next变量。
- 我们首先将pre移动到第一个要反转的节点的前面,即pre->next=left
- 然后将指针cur移动到第一个要反转的节点位置上,即cur=left,
- 然后将 cur->next 赋值给变量next。
- 将cur的下一个节点指向next的下一个节点,即cur->next=next->next
- 然后将next的下一个节点指向pre的下一个节点,即next->next=pre->next
- 然后将next插入到链表头部,即pre->next=next。
- 然后循环往复执行3、4、5、6,直到反转完链表区间内的节点。
下面我们来看一下代码如何实现。
class Solution:
def reverseBetween(self, head, left, right):
#设置哨兵节点,对于链表相关的问题,我们通过设置哨兵节点
#可以省去边界条件的判断
dummynode = ListNode(-1)
#哨兵节点指向链表头
dummynode.next = head
pre = dummynode
#遍历,使得pre指向链表反转部分
#的第一个结点left
for _ in range(left - 1):
pre = pre.next
#cur指向链表反转部分的第一个节点
cur = pre.next
for _ in range(right - left):
next = cur.next
cur.next = next.next
next.next = pre.next
pre.next = next
return dummynode.next
该算法的时间复杂度是O(N),其中 N 是链表总节点数。最多只遍历了链表一次,就完成了反转。空间复杂度是O(1),只用到了常数个变量。