这道题和上面两道相比,是反转链表中更加普通的情况。
解决这道题需要两次循环,第一次求得长度,通过长度分段进行反转。第二次循环分段进行反转,分段反转就是Reverse Linked List II的情况了。
在第二次循环中有两个循环,一个是外层循环,一个是内层循环。内层的逻辑和之前的代码完全一致。外层循环的任务是改变prev(prev = cur)和循环变量。
具体代码如下:
class Solution { public: ListNode* reverseKGroup(ListNode* head, int k) { if (head == NULL || head->next == NULL || k == 1) { return head; } ListNode* dummyHead = new ListNode(0); dummyHead->next = head; ListNode* prev = dummyHead; ListNode* cur = prev->next; ListNode* nex = cur->next; int len = 0; while(cur) { ++len; cur = cur->next; } while(len >= k) { cur = prev->next; nex = cur->next; for(int i=0; i<k-1; ++i) { cur->next = nex->next; nex->next = prev->next; prev->next = nex; nex = cur->next; } prev = cur; len -= k; } ListNode* retHead = dummyHead->next; delete dummyHead; return retHead; } };