算法通关村第二关 | k个一组反转链表 | java

80 阅读1分钟

图片.png

1.头插法

思路:和一般的完整链表头插法思路一样

图片.png

区别在于我们要反转几个链表,以及反转完成一组链表后,对应指针的指向。 反转多少次的问题可以通过链表长度/k来确定。

public ListNode reverseKGroup(ListNode head, int k) {
        ListNode dummy = new ListNode(-1,head);
        ListNode pre = dummy;
        ListNode cur = dummy.next;
        int len = 0;
        while(head!=null){//求链表长度
            len++;
            head = head.next;
        }
        for(int i=0;i<len/k;i++){//反转链表的组数
            for(int j=0;j<k-1;j++){//区间反转
                ListNode next = cur.next;
                cur.next = next.next;
                next.next = pre.next;
                pre.next = next;
            }
            pre = cur;//指向下一组链表的前节点
            cur = cur.next;
        }
        return dummy.next;
    }

2.穿针引线法

思路:也是分组反转链表,将待反转链表抽离,反转完成后再接入原链表。注意的是,反转完成后,该组链表的头节点为end即pre.next=end

e3024148871f6e8d3bd33b1a12e0c57.jpg

public ListNode reverseKGroup(ListNode head, int k) {
        ListNode dummy = new ListNode(-1,head);
        ListNode pre = dummy;
        ListNode end = dummy;
        while(end.next!=null){
            for(int i=0;i<k && end!=null;i++){
                end = end.next;
            }
            if(end==null) break;
            //区间裁剪
            ListNode start = pre.next;
            ListNode next = end.next;
            end.next = null;//区间裁剪
            
            pre.next = reverse(start);//reverse方法执行后等同于pre.next = end;
            pre = start;
            end = start;
            end.next = next;//指向下一组链表
        }
        return dummy.next;
    }
    public static ListNode reverse(ListNode head){//反转链表
        ListNode pre = null;
        ListNode cur = head;
        ListNode next = null;
        while(cur!=null){
            next = cur.next;
            cur.next = pre;
            pre = cur;
            cur = next;
        }
        return pre;
    }