给出一个链表,每 k 个节点一组进行翻转,并返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么将最后剩余节点保持原有顺序。
示例 :
给定这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
思路:
1. 判断是否是空,或者只有一个节点,返回head就可以
2. 然后需要新建一个dummy节点,作为头结点的前一个结点
3. 然后计数从head开始满k就进行反转一次
4. 注意需要协助一个函数功能是反转链表,传入的参数是需要反转的节点的前一个和后一个
5. 这个函数返回的是反转链表中的尾结点,也就是下次反转的头结点的前一个结点
6. 注意反转链表的方法,要提前记录头结点head
public ListNode reverseKGroup(ListNode head, int k) {
if(head == null || head.next == null) return head;
int count = 0;
ListNode dummy = new ListNode(0);
dummy.next = head;
ListNode pre = dummy;
ListNode cur = head;
while(cur != null){
ListNode next = cur.next;
count++;
if(count == k){
pre = reverse(pre, next);
count = 0;
}
cur = next;
}
return dummy.next;
}
public static ListNode reverse(ListNode pre, ListNode end){
if(pre == null || pre.next == null){
return pre;
}
ListNode head = pre.next;
ListNode cur = pre.next.next;
while(cur != end){
ListNode next = cur.next;
cur.next = pre.next;
pre.next = cur;
cur = next;
}
head.next = end;
return head;
}