leetcode题解-25. Reverse Nodes in k-Group

最新推荐文章于 2020-05-28 13:06:26 发布

原创最新推荐文章于 2020-05-28 13:06:26 发布 · 876 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode题解专栏收录该内容

63 篇文章

订阅专栏

本文介绍了一种算法，用于将链表每K个结点进行翻转，不足K个则保持不变。通过定义多个辅助指针实现链表的有效翻转，并提供了完整的Java代码实现。

题意：把原始链表k个k个的反转，如果最后剩余的不到k个结点，那么保持不变。

例子：
给定链表: 1->2->3->4->5
如果 k = 2, 返回链表: 2->1->4->3->5
如果 k = 3, 返回链表: 3->2->1->4->5

分析：前两天十大上有师兄分享17年算法岗面经时提到了本题，因此在做本题的时候也格外认真。晚上洗澡的时候理了一下思路，然后手写代码bug-free，还是很开心的。

题目本身是几道题的综合，因此只要前面的链表题好好做了，包括Reverse Linked List II和Swap Nodes in Pairs。下面讲一下思路：

首先定义几个指针。
老规矩dummyNode指向头节点的前置节点，因为有可能涉及到头节点的操作。
checkNode用来检查是否剩余的有k个节点，当不足k个节点时，直接返回dummyNode.next。
pre、left和right和Reverse Linked List II中的作用相同，用来Reverse节点。

那么对于链表1->2->3->4->5来说，翻转过程如下：
1->2->3->4->5
2->1->3->4->5
3->2->1->4->5

因此代码为：

class Solution {    
    public static ListNode reverseKGroup(ListNode head, int k) {
        if(head == null || head.next == null) return head;
        ListNode dummyNode = new ListNode(0); 
        dummyNode.next = head;
        ListNode pre = dummyNode;
        ListNode left = head;
        ListNode right = head.next;
        ListNode checkNode = head;
        while(true){
            for(int i = 0; i < k; i++){
                if(checkNode == null){
                    return dummyNode.next;
                }
                checkNode = checkNode.next;
            }
            for(int i = 0; i < k - 1; i++){
                left.next = right.next;
                right.next = pre.next;
                pre.next = right;
                right = left.next;
            }
            pre = left;
            left = right;
            // right 有可能是空节点
            if(right == null){
                return dummyNode.next;
            }else{
                right = right.next;
            }
        }
    }

    public static void print(ListNode head){
        while(head != null){
            System.out.println(head.val);
            head = head.next;
        }

    }
    public static void main(String[] args) {
        ListNode l1 = new ListNode(1);
        ListNode l2 = new ListNode(2);
        ListNode l3 = new ListNode(3);
        ListNode l4 = new ListNode(4);
        ListNode l5 = new ListNode(5);
//
//
//
        l1.next = l2;
        l2.next = l3;
        l3.next = l4;
        l4.next = l5;


        print(reverseKGroup(l1, 2));
    }

}