Given the head
of a linked list, rotate the list to the right by k
places.
Example 1:
Input: head = [1,2,3,4,5], k = 2 Output: [4,5,1,2,3]
Example 2:
Input: head = [0,1,2], k = 4 Output: [2,0,1]
Constraints:
- The number of nodes in the list is in the range
[0, 500]
. -100 <= Node.val <= 100
0 <= k <= 2 * 109
题目说给定一个链表,将其向右选择旋转K个位置。所谓向右旋转就是每个节点向右移动,链表尾节点再向右移动就变成链表的头节点。因此可以把链表首尾相连变成一个环,然后再去找向右移动K个位置后新的头节点和尾节点,在新的尾节点和新的头节点之间断开就形成了移动后的新的链表。
假设链表一共有n个节点,那向右移动k个位置后谁是新的头节点和尾节点呢?可以看出倒数第k个节点移动k-1步后就到了尾部,再移动一步就到了链表的头部。因此倒数第k个(正数第n-k+1个)节点就是新的头节点,那么倒数第k+1个(正数第n-k个)节点就是新的尾节点。
另外我们考虑要是k大于等于n该怎么办。很显然把链表移动n(或者n的整数倍)个位置后还是链表本身,因此我们只需考虑k除以n的余数部分。也就是说对于任意k只需移动k%n步就行。
本题解法:把链表尾节点的next指向头节点形成一个环,然后从头节点开始找到第(k%n)个节点也就是新的尾节点,把新的尾节点的next也就是新的头节点存下来,然后再将新的尾节点的next指向空节点,这就形成了移动k个位置后的新链表。
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def rotateRight(self, head: Optional[ListNode], k: int) -> Optional[ListNode]:
if not head or not head.next or k == 0:
return head
cur, cnt = head, 1
while cur.next:
cnt += 1
cur = cur.next
cur.next = head
k %= cnt
for i in range(cnt - k):
head = head.next
cur = cur.next
cur.next = None
return head