猿辅导手撕算法 K个一组翻转链表

哎,算法需要加强加强加强加强!!!!!!!!!!

题目: K 个一组翻转链表
给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。

k 是一个正整数,它的值小于或等于链表的长度。

如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。

示例:

给你这个链表:1->2->3->4->5

当 k = 2 时,应当返回: 2->1->4->3->5

当 k = 3 时,应当返回: 3->2->1->4->5

说明:

你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。

/**
 * 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) {
    
    
        ListNode dummy=new ListNode(0);   //因为头节点会变动,所以需要构造一个伪节点,来指向头节点
        dummy.next=head;

        ListNode pre=dummy;
        ListNode end=dummy;
        while(end.next!=null){
    
    
            for(int i=0;i<k&&end!=null;i++){
    
    
                end=end.next;
            }                  //定位到end节点该在的位置
            if(end==null){
    
        //若在定位的过程中end为null直接退出
                break;
            }
            ListNode next=end.next;   //保存end节点的下一个位置
            ListNode start=pre.next;   //得到翻转的起始节点
            end.next=null;               //得到start-end的链表段
            pre.next=reverse(start);     //翻转链表后,用pre指向新链表的头
            start.next=next;         //旧链表的start节点现在是新链表的最后一个节点,所以指向下一段链表(next)
            pre=start;     //开始下一段循环
            end=pre;
        }
        return dummy.next;        //返回头节点
    }

    private static ListNode reverse(ListNode head)
    {
    
    
        ListNode pre=null;
        ListNode cur=head;
        while(cur!=null)
        {
    
    
            ListNode next=cur.next;
            cur.next=pre;
            pre=cur;
            cur=next;
        }
        return pre;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_42684418/article/details/107928600
今日推荐