LeetCode 61. Rotate List - 链表(Linked List)系列题12

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值
>