打卡群刷题总结0715——旋转链表

题目：61. 旋转链表

链接：https://leetcode-cn.com/problems/rotate-list

给定一个链表，旋转链表，将链表每个节点向右移动 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

解题：

1、根据题意，旋转k（k>=length）次等同于旋转k % length次。那么可以先更新k，再进行旋转，方法如下：

1）得到链表长度，记为length。

2）更新k为k % length。

3）找到倒数第(k+1)个节点，记为p。（注意：是倒数第k+1个节点，而不是倒数第k个节点。）

4）旋转链表：将头结点和尾节点记为head和tail，tail.next指向head，head指向p.next，p.next修改为空。

代码：

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution(object):
    def rotateRight(self, head, k):
        """
        :type head: ListNode
        :type k: int
        :rtype: ListNode
        """
        if not head or not head.next:
            return head

        # 节点个数
        length = 1
        p = head
        while p.next:
            length += 1
            p = p.next

        k = k % length
        if k == 0:
            return head

        # 找到前一个节点
        q = head
        count = 1
        while count < length - k:
            count += 1
            q = q.next

        # 旋转
        p.next = head
        head = q.next
        q.next = None
        return head