92. 反转链表 II
题目链接
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reverse-linked-list-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]
提示:
-
链表中节点数目为 n
-
1 <= n <= 500
-
-500 <= Node.val <= 500
-
1 <= left <= right <= n
-
进阶: 你可以使用一趟扫描完成反转吗?
题目分析
- 根据题目描述反转链表里部分元素
- 把链表分为三部分,需要反转的前半段、反转的部分、需要反转的后半段
- 需要反转的部分存放在stack中,利用后进先出的原理反转链表
- 最后把需要反转的前半段、反转后的部分、需要反转的后半段,这三部分拼接起来
代码
/**
* 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* reverseBetween(ListNode* head, int left, int right) {
stack<ListNode*> listStack;
ListNode* p = head;
ListNode* front = nullptr;
ListNode* rear = nullptr;
int index = 1;
while (p) {
if (index == left - 1) front = p;
if (index == right + 1) rear = p;
if (index >= left && index <= right) listStack.push(p);
index++;
p = p->next;
}
if (front == nullptr) {
head = listStack.top();
listStack.pop();
front = head;
}
while (!listStack.empty()) {
front->next = listStack.top();
listStack.pop();
front = front->next;
}
front->next = rear;
return head;
}
};