题目描述
Given a linked list, reverse the nodes of a linked list k at a time and return its modified list.
k is a positive integer and is less than or equal to the length of the linked list. If the number of nodes is not a multiple of k then left-out nodes in the end should remain as it is.
Example:
Given this linked list: 1->2->3->4->5
For k = 2, you should return: 2->1->4->3->5
For k = 3, you should return: 3->2->1->4->5
Note:
Only constant extra memory is allowed.
You may not alter the values in the list’s nodes, only nodes itself may be changed.
解题代码
class Solution {
public:
ListNode *reverseKGroup(ListNode *head, int k) {
if(head==NULL||k==1) return head;
int num=0;
ListNode *preheader = new ListNode(-1);
preheader->next = head;
ListNode *cur = preheader, *nex, *pre = preheader;
while(cur = cur->next)
num++;
while(num>=k) {
cur = pre->next;
nex = cur->next;
for(int i=1;i<k;++i) {
cur->next=nex->next;
nex->next=pre->next;
pre->next=nex;
nex=cur->next;
}
pre = cur;
num-=k;
}
return preheader->next;
}
};
解题思路
- 如果链表为空或只有一个元素则返回。
- 如果链表大于等于2个元素,在链表头部增加一个新节点(pre),依次遍历链表,使得pre指向新遍历的节点,并将新遍历节点的指针指向已经遍历过的链表。
- 详情见链接:https://leetcode.com/problems/reverse-nodes-in-k-group/discuss/11457/20-line-iterative-C++-solution
- 因为每k个数一个小组,所以需要先遍历整个链表得到整个链表的长度,才能知道最后什么不用再反转了。
解题代码
class Solution
{
public:
ListNode* reverse(ListNode* first, ListNode* last)
{
ListNode* prev = last;
while ( first != last )
{
auto tmp = first->next;
first->next = prev;
prev = first;
first = tmp;
}
return prev;
}
ListNode* reverseKGroup(ListNode* head, int k)
{
auto node=head;
for (int i=0; i < k; ++i)
{
if ( ! node )
return head; // nothing to do list too sort
node = node->next;
}
auto new_head = reverse( head, node);
head->next = reverseKGroup( node, k);
return new_head;
}
};
解题思路
reverse函数是进行反转的,第二个参数指向的节点是翻转节点的下一个节点。