ONT60 旋转链表 思路分享

题干链接：ONT60 旋转链表

​

这道题是反转链表题的pro升级版，但比反转链表略微复杂一些。如果有做过旋转数组那道题（链接在这里：https://blog.youkuaiyun.com/wyd_333/article/details/126712919，但当时刷这道题的时候我用的是二分法，其实是复杂化了），思路很容易想到：先将整个链表整体反转，再分别反转前k个和剩下的。

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

这里反转的这个操作，复用的就是“反转链表”那道题中的思路。这里记录一下AC时容易遇到的几个细节坑：

1. reverse()方法应有返回值。如果调用时候传入的是头节点head，那么作为引用传递，在反转之后head的值是会被改变的，会变成链表的尾部，如果不记录新的头节点，后序就没法再对反转之后的链表进行操作了。可以在调用方法之前记录，也可以通过返回值的方式记录新头节点，推荐返回值的方式。

2.k的值需要处理。在找第k个元素时，k有边界情况要处理。如k > 链表长度。遇到这种情况是直接舍弃，还是轮回，要看题意。这道题题意要求是通过轮回的方式来处理的（WA几次看未通过用例就可以知道了）。因此 k = k % len 是必须的。

3.理清楚谁是头节点，谁是尾节点。这里涉及多次反转，反转之后谁是头谁是尾，要把谁接到谁之后？这些问题是容易出错的，需要想清楚。

代码：

import java.util.*;

/*
 * public class ListNode {
 *   int val;
 *   ListNode next = null;
 *   public ListNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param head ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    public ListNode rotateLinkedList (ListNode head, int k) {
        if(head == null) {
            return null;
        }
        int len = getLen(head);
        k = k % len;

        ListNode afterReverseHead = reverse(head);
        // testPrint(afterReverseHead);
        ListNode cur = afterReverseHead;
        while(k > 1) {
            k--;
            cur = cur.next;
        }

        System.out.println(cur.val);
        ListNode newHead = cur.next;
        cur.next = null;
        ListNode preHead = reverse(afterReverseHead);
        ListNode postHead = reverse(newHead);

        afterReverseHead.next = postHead;
        return preHead;
    }

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

    public int getLen(ListNode head) {
        int count = 0;
        while(head != null) {
            head = head.next;
            count++;
        }
        return count;
    }
}