92. 反转链表 II

https://leetcode.cn/problems/reverse-linked-list-ii/description/

给你单链表的头指针 head 和两个整数 left 和 right ,其中 left <= right 。请你反转从位置 left 到位置 right 的链表节点,返回 反转后的链表 。

示例 1:

输入:head = [1,2,3,4,5], left = 2, right = 4
输出:[1,4,3,2,5]

示例 2:

输入:head = [5], left = 1, right = 1
输出:[5]

提示:

链表中节点数目为 n
1 <= n <= 500
-500 <= Node.val <= 500
1 <= left <= right <= n

进阶: 你可以使用一趟扫描完成反转吗?

解题思路

这题是 206. 反转链表 的衍生题目,可以理解为将链表根据left 和right 分成三段,将中间那一段进行反转

代码

type ListNode struct {
	Val  int
	Next *ListNode
}

func reverseBetween(head *ListNode, left int, right int) *ListNode {
	var preHead, r *ListNode
	// preHead 指向头节点
	preHead = &ListNode{
		Val:  0,
		Next: head,
	}
	// pre 用来记录遍历的上一个节点
	pre := preHead
	// tmp 用来记录需要反转的链表
	tmp := preHead
	for head != nil {
		left--
		right--
		// 到达需要反转列表的头节点
		if left == 0 {
			// 切断列表,记录反转列表
			pre.Next = nil
			tmp = head
		}
		// 到达反转列表的尾部
		if right == 0 {
			// 切断列表,记录剩余部分
			r = head.Next
			head.Next = nil
		}
		pre = head
		head = head.Next
	}
	// 反转链表
	tmp = reverse(tmp)
	res := preHead
	// 拼接链表
	for preHead.Next != nil {
		preHead = preHead.Next
	}
	preHead.Next = tmp
	for tmp.Next != nil {
		tmp = tmp.Next
	}
	tmp.Next = r
	return res.Next
}

func reverse(head *ListNode) *ListNode {
	if head == nil || head.Next == nil {
		return head
	}
	var pre *ListNode
	for head.Next != nil {
		tmp := head.Next
		head.Next = pre
		pre = head
		head = tmp
	}
	head.Next = pre
	return head
}