题目
给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
进阶:
你可以设计一个只使用常数额外空间的算法来解决此问题吗?
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
题解
首先要了解到链表天然具有递归属性,因为一个链表可以看作是一个节点连着另一个链表,即递归中更小的子问题。
对于k个一组反转链表,因此去除给定链表的前k个节点,剩余的节点组成的链表依旧满足k个一组反转链表这个条件,这就是这个题目的子问题。所以我们需要做的就是找到解决子问题的方法,然后递归处理就好。
在主函数中,首先判断当前头节点开始的链表是否还有k个节点。若有,则通过reverse函数得到当前组的翻转链表,newHead为当前组翻转后的头节点,b节点为下一组的头节点,因此可以作为下一个子问题的头节点。
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode reverseKGroup(ListNode head, int k) {
if(head == null) return null;
ListNode a, b;
a = b = head;
for(int i = 0; i < k; i++){
if(b == null) return head; //剩余节点不足k个,无需翻转直接返回
b = b.next;
}
// 反转前k个节点
ListNode newHead = reverse(a, b);
// 此时a节点是末尾节点,要连上以b节点为首的后续链表,并对后面的节点也递归翻转
a.next = reverseKGroup(b, k);
return newHead;
}
private ListNode reverse(ListNode a, ListNode b){ // 反转[a,b)之间的链表,左闭右开
ListNode pre, cur, nxt;
pre = null;
cur = a;
nxt = a;
while(cur != b){
nxt = cur.next;
cur.next = pre;
pre = cur;
cur = nxt;
}
return pre;
}
}