Python描述 LeetCode 61. 旋转链表

Posted 2022-07-31 亓官劼

Python描述 LeetCode 61. 旋转链表

  大家好，我是亓官劼（qí guān jié ），在【亓官劼】公众号、CSDN、GitHub、B站等平台分享一些技术博文，主要包括前端开发、python后端开发、小程序开发、数据结构与算法、docker、Linux常用运维、NLP等相关技术博文，时光荏苒，未来可期，加油~

  如果喜欢博主的文章可以关注博主的个人公众号【亓官劼】(qí guān jié)，里面的文章更全更新更快。如果有需要找博主的话可以在公众号后台留言，我会尽快回复消息.

---

本文原创为【亓官劼】（qí guān jié ），请大家支持原创，部分平台一直在恶意盗取博主的文章！！！ 全部文章请关注微信公众号【亓官劼】。

题目

给你一个链表的头节点 head ，旋转链表，将链表每个节点向右移动 k 个位置。

示例 1：

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-SJIaR6Vv-1655343200285)(https://assets.leetcode.com/uploads/2020/11/13/rotate1.jpg)]

输入：head = [1,2,3,4,5], k = 2
输出：[4,5,1,2,3]

示例 2：

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-5oPBES9q-1655343200286)(https://assets.leetcode.com/uploads/2020/11/13/roate2.jpg)]

输入：head = [0,1,2], k = 4
输出：[2,0,1]

提示：

• 链表中节点的数目在范围 [0, 500] 内
• -100 <= Node.val <= 100
• 0 <= k <= 2 * 109

Python描述

快慢指针，快指针先走k%len步，然后慢指针再走，然后交换一下next即可

# 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 k == 0 or head == None:
            return head
        tmp_head = ListNode(-1,head)
        p = tmp_head.next

        n = 0
        while p:
            n += 1
            p = p.next
        k = k % n
        
        p = tmp_head.next
        while k:
            p = p.next
            k -= 1
        
        q = tmp_head.next
        while p.next and q.next:
            p = p.next
            q = q.next
        
        p.next = tmp_head.next
        tmp_head.next = q.next
        q.next = None
        return tmp_head.next