LeetCode#25 K 个一组翻转链表

2020-07-04  本文已影响0人  Android_ZzT

题目:

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

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

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

示例:

给定这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5

代码:

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) { //如果end下一个还有节点,进入循环
            for (int i=0; i<k && end != null; i++) { //end 移动到这组的最后
                end = end.next;
            }
            if (end == null) break; //如果长度不够一组,跳出循环
            
            ListNode start = pre.next; //start为要反转的 head 
            ListNode next = end.next; //end 的下一个节点保存到 next
            
            end.next = null; //断开组与后面的连接
            pre.next = reverse(start); //连接反转过的链表
            
            start.next = next; //将反转过的组与后面节点相连
            
            pre = start; //重置 pre 和 end,此时前面一组节点已经反转好,准备进入下一组
            end = pre;
        }
        return dummy.next;
    }
    
    private 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;
    }
}
上一篇 下一篇

猜你喜欢

热点阅读