老卫带你学---leetcode刷题(61. 旋转链表)

最新推荐文章于 2023-11-23 23:34:10 发布

老卫带你学

最新推荐文章于 2023-11-23 23:34:10 发布

阅读量55

点赞数

分类专栏： leetcode刷题文章标签： leetcode

本文链接：https://blog.csdn.net/yixieling4397/article/details/134001649

版权

leetcode刷题专栏收录该内容

120 篇文章

订阅专栏

本文介绍了如何通过创建环形链表并调整节点位置来解决旋转链表的问题，给出了Python代码实现，适用于给定链表节点和旋转步数k的情况。

摘要由CSDN通过智能技术生成

61. 旋转链表

问题：

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

示例 1：


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

示例 2：


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

提示：

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

解决：

先将给定的链表连接成环，然后将指定位置断开

首先计算出链表的长度 n，并找到该链表的末尾节点，将其与头节点相连。这样就得到了闭合为环的链表。然后我们找到新链表的最后一个节点n-k%n 将当前闭合为环的链表断开，即可得到我们所需要的结果

class Solution:
    def rotateRight(self, head: Optional[ListNode], k: int) -> Optional[ListNode]:
        if k == 0 or not head or not head.next:
            return head
        
        n = 1
        cur = head
        while cur.next:
            cur = cur.next
            n += 1
        
        if (add := n - k % n) == n:
            return head
        
        cur.next = head
        while add:
            cur = cur.next
            add -= 1
        
        ret = cur.next
        cur.next = None
        return ret