1. 程式人生 > >25. k個一組翻轉連結串列

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) {}
 * };
 */
class Solution { public: ListNode* reverseKGroup(ListNode* head, int k) { if(k <= 1 || !head) { return head; } ListNode dummy(0); dummy.next = head; ListNode* p = &dummy; ListNode* prev = &dummy; while(p) { prev =
p; for(int i = 0; i < k; i++) { p = p->next; if(!p) { return dummy.next; } } p = reverse(prev, p->next); } return dummy.next; } ListNode* reverse(ListNode* prev, ListNode*
end) { ListNode* p = prev->next; while(p->next != end) { ListNode* n = p->next; p->next = n->next; n->next = prev->next; prev->next = n; } return p; } };