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

Posted 2022-11-27 BridgeGeorge

题目

链接：https://leetcode-cn.com/problems/reverse-nodes-in-k-group/

给你一个链表，每 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;