算法系列——k个一组翻转链表

题目

链接：leetcode-cn.com/problems/re…

给你一个链表，每 k 个节点一组进行翻转，请你返回翻转后的链表。
k 是一个正整数，它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。

思路

核心过程就是 反转 [a,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 head;
        }
        ListNode a=head;
        ListNode b=head;
        for(int i=0;i<k;i++){
            if(b==null){
                return head;
            }
            b=b.next;
        }
        //得到链表头
        ListNode newHead=reverse(a,b);
        //a 链接新的 反转后的链表头
        a.next=reverseKGroup(b,k);
        return newHead;
    }

    //反转 [a,b) 区间的链表
    private ListNode reverse(ListNode a,ListNode b){
        ListNode prev=null;
        ListNode cur=a;
        ListNode next=null;
        while(cur!=b){
            next=cur.next;
            cur.next=prev;
            prev=cur;
            cur=next;
        }
        return prev;
    }
}