题解 | 链表中的节点每k个一组翻转
链表中的节点每k个一组翻转
https://www.nowcoder.com/practice/b49c3dc907814e9bbfa8437c251b028e
/**
* struct ListNode {
* int val;
* struct ListNode *next;
* ListNode(int x) : val(x), next(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param head ListNode类
* @param k int整型
* @return ListNode类
*/
ListNode* reverse(ListNode* pre, int k){
// 给出一组的前一个节点,然后翻转该组,返回结果指向翻转后的头结点。
ListNode* cur = pre->next;
for(int m=1; m<k; m++){
ListNode* temp = cur->next;
cur->next = temp->next;
temp->next = pre->next;
pre->next = temp;
}
return pre->next;
}
ListNode* reverseKGroup(ListNode* head, int k) {
// write code here
if(k <= 1) return head;
if(head == NULL) return head;
ListNode* dummy = new ListNode(0);
dummy->next = head;
ListNode* prev = dummy;
ListNode* cur = head;
int len = 0;
while(cur){
cur = cur->next;
len++;
}
int s = len / k;
for(int i = 0; i < s; i++){
if (i == 0){
dummy->next = reverse(prev, k);
prev->next = dummy->next;
}
else{
prev->next = reverse(prev, k);
}
for(int j = 0; j < k; j++){
prev = prev->next;
}
}
return dummy->next;
}
};
类似BM2的解题思路。构造一个reverse函数快速实现组内的翻转,然后添加k组的思路即可。
华为HUAWEI工作强度 1383人发布
