leetcode链表之旋转链表

61、旋转链表

文章出现的代码都是python3

题目

给你一个链表的头节点 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]

思路

旋转链表，这个时候很容易想到环，然后每个节点向右移动k个位置，和之前查找倒数第N个结点很像，需要在其位置进行断开操作。

所以可以先计算链表的长度，顺便将头尾结点连接成环，此时找到新的尾结点进行断开即可。

代码：

class Solution:
    def rotateRight(self, head: Optional[ListNode], k: int) -> Optional[ListNode]:
        # 当k为0，或者链表为空，或者链表只有一个元素的时候，这个时候其实没必要旋转链表
        if not k or not head or not head.next: return head
        # 第一步，计算链表长度，并记录尾结点
        cur = head
        length = 1
        while cur.next:
            length += 1
            cur = cur.next
        # 如果k是长度的整数倍，其实也没必要旋转链表
        if k % length == 0: return head
        # 此时cur的next指针指向的就是尾结点，此时将其next指向头结点就形成了环
        cur.next = head
        # 第二步，找到需要断开结点的前驱结点
        k = length - (k % length)
        while k:
            cur = cur.next
            k -= 1
        # 第三步，由于从原来的尾结点进行循环，循环k次，此时的cur就是要断开结点的前驱结点，断开即可
        newHead = cur.next
        cur.next = None
        return newHead