Reverse Nodes in k-Group
Difficulty:Hard
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.
You may not alter the values in the nodes, only nodes itself may be changed.
Only constant memory is allowed.
For 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
这道题就是要根据k对链表进行分段逆置,比如当链表为1->2->3->4->5,k=2时,就是对1->2,3->4,分别进行逆置得到2->1,4->3,最后的5因为不足2个元素所以不进行逆置,所以结果为2->1->4->3->5。
实现的方法就是先从头开始选前k个进行逆置(设链表为1->2->3->4->5,k=2时,第一步结果就会是2->1->3->4->5),然后再往后对连续k个进行逆置(2->1->4->3->5),一直重复这个步骤,直到逆置完整个链表或剩余的节点不足k个时,就得到结果。其中对k个节点进行的逆置就是一般链表的逆置,但是注意题目要求空间负责度为常数,所以只能在原链表上进行逆置。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
ListNode* reverseKGroup(ListNode* head, int k) {
if (head == NULL) //链表为空
return NULL;
ListNode *first = head, *last = NULL, *tmp = NULL, *current, *tmpFirst; //first和last就是当前选取的k个节点的头和尾,tmp、current、tmpFirst用于逆置k个节点
ListNode *lastReverse=NULL, *nextReverse, *result=head; //lastReverse为上一次逆置完k个节点后的末尾,nextReverse为下一次要进行选取的开头
while (1){
if (first == NULL)//first为NULL说明链表刚好是k的整数倍,正好逆置完
break;
last = first;
for (int i = 0; i < k - 1; i++){ //选取k个节点
if (last->next != NULL)
last = last->next;
else{
return result;
}
}
nextReverse = last->next; //last的下一个节点就是下一次选取的起点
tmpFirst = first;
tmp = NULL;
int m = 0;
do{
current = tmpFirst;
if (tmpFirst->next!=NULL)
tmpFirst = tmpFirst->next;
current->next = tmp;
tmp = current;
m++;
} while (m < k); //这个do while就是对k个节点进行逆置
first->next = nextReverse; //逆置完后first就变成了k个节点里最后的节点,所以指向后面剩余的链表
if (lastReverse!=NULL) //如果lastReverse不为空,也就是当前不是第一次逆置,则上一次逆置的最后一个借点指向last
lastReverse->next = last;
else //第一次逆置后,方法返回值就是第一次逆置后的last,也就是完成逆置后的第一个节点
result = last;
lastReverse = first; //当前k个逆置完成后,first就成为了上一次逆置的最后一个节点
first = nextReverse; //把下一次逆置的开始节点赋值给first
}
return result;
}