Leetcode 92 反转链表II
一、题目
给你单链表的头指针 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]
二、解法
我们之前在 Leetcode 206 反转链表 里学会了如何反转链表,这次的题目要求反转一个区间内的链表,其他部分保持不变。
最直观的思路是用两次遍历找到处于left位置的结点的上一个结点,和处于right位置的结点的下一个结点,以及处于left位置的结点。
记录下它们三个的位置,调用reverseList函数反转区间,再把反转后的链表和头尾进行拼接。
考虑到有可能出现待反转部分的left为链表第一个位置的情况,所以使用虚拟头结点的技巧来简化边界判断。
以示例1为例:
要记录的三个位置如下
记录1号结点是为了方便把反转后的链表和前面的链表元素连接起来。
记录2号结点是为了传入参数给调用的反转函数。
记录5号结点是因为反转函数是从头到尾进行反转,需要先断开4号结点与后面的链表元素的连接,待反转结束后再把反转后的链表与后面的链表元素连接起来。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* reverseList(ListNode* head){
if(head==nullptr || head->next==nullptr) return head;
ListNode* p=head->next,*q=reverseList(head->next);
head->next=nullptr;
p->next=head;
return q;
}
ListNode* reverseBetween(ListNode* head, int left, int right) {
if(head==nullptr || head->next==nullptr) return head;
ListNode virnode(0,head),*p=&virnode,*q;
for(int i=1;i<left;i++){
p=p->next;
}
q=p->next;
for(int i=left;i<right;i++){
q=q->next;
}
ListNode* temp1=p->next;
ListNode* temp2=q->next;
q->next=nullptr;
p->next=reverseList(p->next);
temp1->next=temp2;
return virnode.next;
}
};