Leetcode 61. 旋转链表（Rotate List）

最新推荐文章于 2021-06-20 22:31:22 发布

原创最新推荐文章于 2021-06-20 22:31:22 发布 · 125 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法 #leetcode

Leetcode 专栏收录该内容

595 篇文章

订阅专栏

本文提供了一种解决LeetCode上第61题“旋转链表”的算法思路，通过双指针法找到链表旋转的位置，特别关注于处理k值大于链表长度的情况，确保链表正确旋转。

Leetcode 61. 旋转链表

1 题目描述（Leetcode题目链接）

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

输入: 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

输入: 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

2 题解

题目的意思就是把后 $k$ 个节点翻到前面去，这样的话只需要用双指针找到翻的位置就好了，但是 $k$ 的大小可能比链表长度大，这样就需要先获取链表长度，并找到最后一个节点，再根据 $k$ 值找到翻的位置就好了。

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

class Solution:
    def rotateRight(self, head: ListNode, k: int) -> ListNode:
        if not head:
            return None
            
        length = 1 #链表长度
        last = head #链表最后一个节点
        while last.next:
            length += 1
            last = last.next

        k = k % length
        if k == 0: return head #如果k是长度的整数倍，则直接返回head

        step = length - k - 1
        temp = head #temp是要翻的位置
        while step > 0: 
            temp = temp.next
            step -= 1

        newHead = temp.next
        temp.next = None
        last.next = head
        return newHead