这道题的大意是说,给一个链表,再给一个整型数字K,要实现这个链表每K个数进行一次反转,整条链表反转的题目比较熟悉,但是,一个链表分成好几组进行反转其实大体思路是一样的,但是每组反转结束之后,应该如何变化指针进行下一次反转,这是这道题的重点。
按照以前链表反转的思路,我们首先要创建一个头节点,用来记录链表头节点的位置。
其次,设置计数值,将每组的头尾节点分别记录,这块要注意,最好要记录每组头节点的前一个节点,因为每组反转后,每组原来的最后一个节点要变为第一个节点,也就是要让上一组的最后一个节点指向它,所以,此处要记录的是每组第一个元素的前一个元素的地址: dummy -> 1 -> 2 -> 3 -> 4 ->NULL,K=2,第一次记录dummy和2,第二次记录2和4,从而实现各组反转。这块想通了代码也就比较简单了。
以下是我的代码:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* reverseKGroup(ListNode* head, int k) {
if (!head || k<2) {
return head;
}
ListNode* dummy = new ListNode(0);
ListNode* prev = dummy, *cur = head;
dummy->next = head;
int count = 1;
while (cur) {
if (count%k) {
cur = cur->next;
} else {
prev = reverseKNodes(prev, cur->next);
cur = prev->next;
}
++count;
}
return dummy->next;
}
ListNode* reverseKNodes(ListNode* prev, ListNode* next) {
ListNode* head = prev->next;
ListNode* cur = head->next;
while (cur != next) {
head->next = cur->next;
cur->next = prev->next;
prev->next = cur;
cur = head->next;
}
return head;
}
};