Leetcode 25 K个一组翻转链表
程序员文章站
2022-03-18 09:33:19
...
1.题目
2.解法
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode reverseKGroup(ListNode head, int k) {
if (head == null || k <= 0) return head;
// 交换位置从前往后,所以需要记住head
ListNode pre = head;
int count = 0;
while(pre != null && count != k) {
count++;
pre = pre.next;
}
if (count == k) {
pre = reverseKGroup(pre, k);
ListNode next = null;
while(count != 0) {
// 你要指向后面排好的头结点
next = head.next;
head.next = pre;
pre = head;
head = next;
count--;
}
head = pre;
}
// 如果不等于k就需要排列了
return head;
}
}
3.思考
1、重复的部分可以用递归来做
2、首先判断每组能不能有k个元素,如果有k个元素那么就排列,递归的顺序是后面先排,排完前面排
head.next指向上次排完的头结点
3、多观察例子,发现其中的规律