给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
示例 :
给定这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
说明 :
你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reverse-nodes-in-k-group
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法一:指针+递归
先找出是否有k个可供翻转,若不够则直接返回,接着翻转这k个,最后剩下的部分由递归完成,部分参考了陈越姥姥数据结构MOOC的思路。
class Solution {
public:
ListNode* reverseKGroup(ListNode* head, int k) {
if (k < 2) return head;
if (head == NULL || head->next == NULL) return head;
ListNode* dummy = new ListNode(-1);
dummy->next = head;
ListNode* left, * right,* rest;
right = dummy;
int i;
for (i = 0; i < k; i++) {
right = right->next;
if (right == NULL) {
return head;
}
}
left = dummy;
right = left->next;
rest = right->next;
for (i = 1; i < k; i++) {
left = right;
right = rest;
if(rest!=NULL) rest = rest->next;
right->next = left;
}
rest = reverseKGroup(rest, k);
dummy->next->next = rest;
return right;
}
};
方法二:栈+递归
还有一种方法就是把链表的内容都放进栈里,再一个一个弹出来重新组成链表,不知道是否符合“只能使用常数的额外空间”这一条件。
/**
* 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) {
int i;
stack<int> store;
ListNode* p=head;
for (i=0;i<k;i++){
if(p==NULL) return head;
else {
store.push(p->val);
p=p->next;
}
}
ListNode* start=new ListNode(0);
ListNode* q=start;
for(i=0;i<k;i++){
q->next=new ListNode(store.top());
q=q->next;
store.pop();
}
q->next=reverseKGroup(p, k);
return start->next;
}
};