题目描述
给你一个链表的头节点 head ,旋转链表,将链表每个节点向右移动 k 个位置。
提示:
链表中节点的数目在范围 [0, 500] 内
-100 <= Node.val <= 100
0 <= k <= 2 * 109
解题思路
记给定链表的长度为 n,注意到当向右移动的次数 k≥n 时,我们仅需要向右移动 k%n 次即可。因为每 n 次移动都会让链表变为原状。这样我们可以知道,新链表的最后一个节点为原链表的第 (n−1)−(k%n) 个节点(从 0 开始计数)。
这样,我们可以先将给定的链表连接成环,然后将指定位置断开。
具体代码中,我们首先计算出链表的长度 n,并找到该链表的末尾节点,将其与头节点相连。这样就得到了闭合为环的链表。然后我们找到新链表的最后一个节点(即原链表的第 (n−1)−(k%n) 个节点),将当前闭合为环的链表断开,即可得到我们所需要的结果。
注意,当链表长度不大于 1,或者 k 为 n 的倍数时,新链表将与原链表相同,我们无需处理直接返回头节点。
代码实现
public static ListNode rotateRight(ListNode head, int k) {
if(head == null) return null;
int len = getLength(head);
k = k % len;
if(k == 0) return head;
int breakStep = 1;
ListNode breakPoint = head, cur = head, newHead ;
while (breakStep < len ) {
cur = cur.next;
breakStep++;
if (breakStep == len - k ) {
breakPoint = cur;
}
}
newHead = breakPoint.next == null ? breakPoint: breakPoint.next;
breakPoint.next = null;
cur.next = head;
return newHead;
}
private static int getLength(ListNode head) {
int r = 0;
while (head != null) {
head = head.next;
r++;
}
return r;
}