61. 旋转链表(python)

最新推荐文章于 2024-07-20 12:30:43 发布

Briwisdom

最新推荐文章于 2024-07-20 12:30:43 发布

阅读量328

点赞数

分类专栏： LeetCode/竞赛/算法分析文章标签： python

本文链接：https://blog.csdn.net/u010420283/article/details/84659317

版权

LeetCode/竞赛/算法分析专栏收录该内容

57 篇文章 0 订阅

订阅专栏

题目描述（中等）

给定一个链表，旋转链表，将链表每个节点向右移动 k 个位置，其中 k 是非负数。

示例 1:

输入: 1->2->3->4->5->NULL, k = 2
输出: 4->5->1->2->3->NULL
解释:
向右旋转 1 步: 5->1->2->3->4->NULL
向右旋转 2 步: 4->5->1->2->3->NULL

示例 2:

输入: 0->1->2->NULL, k = 4
输出: 2->0->1->NULL
解释:
向右旋转 1 步: 2->0->1->NULL
向右旋转 2 步: 1->2->0->NULL
向右旋转 3 步: 0->1->2->NULL
向右旋转 4 步: 2->0->1->NULL

`思路分析`

如果不考虑链表这一特殊情景，只看每个节点的数字，我们可以看到实际上就是数组内每个位置数字的移动问题，当移动到末尾时候，又重新从0开始移动，所以，本题的关键是设置一个判断：如何处理移动到末尾又从0开始循环移动的问题。

观察可知当移动的步数 k 恰好是数组的长度时候，该位置数字位置不变，所以用 k 除以数组的长度就能够解决尾部循环的问题。根据LeetCode关于链表指针的特殊定义，python3代码实现如下：

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def rotateRight(self, head, k):
        """
        :type head: ListNode
        :type k: int
        :rtype: ListNode
        """
        if not head:
            return         
        res=[]
        while head:
            res.append(head.val)
            head=head.next
        K = k%len(res)
        res = res[-K:] + res[:-K]
        head = h = ListNode(None)
        for i in res:
            head.next = ListNode(i)
            head = head.next
        return h.next