25. K 个一组翻转链表
给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
示例:
给你这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
说明:
你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
#include <iostream>
#include <utility>
using namespace std;
struct ListNode {
int val;ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};
class Solution {
public:
//翻转一个子链表,并且返回新的头和尾
pair<ListNode*,ListNode*> myReverse(ListNode* head,ListNode* tail){
ListNode *prev = tail->next;
ListNode *p = head;
while(prev!=tail){
ListNode *nex = p->next;
p->next = prev;
prev = p;
p = nex;
}
return {tail, head};
}
ListNode* reverseKGroup(ListNode* head, int k) {
ListNode *hair = new ListNode(0);
hair->next = head;
ListNode *pre = hair;
while(head){
ListNode *tail = pre;
//结束条件
//查看剩余部分长度是否大于等于k
for (int i = 0; i < k;i++){
tail = tail->next;
if(!tail){
return hair->next;
}
}
ListNode *nex = tail->next;
pair<ListNode*,ListNode*> result = myReverse(head, tail);
head = result.first;
tail = result.second;
pre->next = head;
tail->next = nex;
pre = tail;
head = tail->next;
}
return hair->next;
}
};
/*
复杂度分析
时间复杂度:O(n),其中 n 为链表的长度。
head 指针会在 O(n/k) 个结点上停留,每次停留需要进行一次 O(k) 的翻转操作。
空间复杂度:O(1),我们只需要建立常数个变量。
*/
作者:LeetCode-Solution
链接
来源:力扣(LeetCode)