给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
示例:
给你这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
说明:
你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reverse-nodes-in-k-group
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
// 分为两步 1.首先判断是否可以翻转
/ 如果可以翻转,首先翻转first->kNode这一段
// 然后再翻转kNode->next 到下一个kNode这一段
class Solution {
public:
ListNode* reverseKGroup(ListNode* head, int k) {
if(k<=1 || head == NULL || head->next == NULL) return head;
ListNode* first = head;
ListNode* kNode = head;
for(int i = k; i > 1 && kNode !=NULL; i--){
kNode = kNode->next;
}
if(kNode == NULL) return head; //判断是否可以翻转
kNode = head;
ListNode* result = new ListNode(1);
ListNode* pre = result;
while(kNode!= NULL) { //遍历,知道链表末尾
int num = k;
while(num >1 && kNode!=NULL) { // 大于1是因为要保留第K个节点
kNode = kNode->next;
num--;
}
if(kNode == NULL) break;//为空说明不够k个不能进行发转
ListNode* tp = first;
ListNode* tp2 = kNode->next;
pre->next = getReverseData(first, kNode);
pre = tp; // 更新pre因为再getReverse中已经改变了first
first = tp2;
kNode = tp2; // 更新kNode,因为再getReverse中已经改变了kNode的next
}
return result->next;
}
ListNode* getReverseData(ListNode* first, ListNode* kNode) {
ListNode* next = first->next;
first->next = kNode->next; //用于翻转first到kNode这一段
while(next!=NULL && next != kNode) {
ListNode* tmp = next->next;
next->next = first;
first= next;
next = tmp;
}
if(next !=NULL) next->next = first;
return next;
}
};