61. 旋转链表
原始题目链接:https://leetcode-cn.com/problems/rotate-list/
给你一个链表的头节点 head ,旋转链表,将链表每个节点向右移动 k 个位置。
解题思路:
用闭环的方法是比较巧妙的,先将链表封闭成环,然后找新链表的尾结点的索引,从索引出断开(next域置空),断开前,索引处的下一个节点就是新链表的头结点直接返回即可。找索引(从0开始计算)的位置用一个计算公式,计链表的长度为n:(n
- 1)- (k % n),特别注意的是当k对n求余为0的时候表示又旋转成了原始链表,所以直接返回head即可,具体实现看代码及注释。
代码实现:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def rotateRight(self, head: ListNode, k: int) -> ListNode:
if head is None or head.next is None:
return head
# 计算链表的长度,n取0还是1有个小技巧,取1可以继续利用cur_node节点
n = 1
cur_node = head
while cur_node.next:
n += 1
cur_node = cur_node.next
# 如果链表为空,或者只有一个节点,或者k是n的整数倍即相当于不用旋转
# 直接返回原始链表的头结点
if k % n == 0:
return head
# 新的链表的尾部位置,节点的索引从0开始计算
new_last_node_index = (n - 1) - k % n
# 把原始链表的尾部节点连接到头结点,形成闭环
# 在计算链表长度时候,退出循环后,cur_node指向了最后一个节点
cur_node.next = head
# 在新链表的尾部断开链表,索引从0开始计算
for _ in range(new_last_node_index + 1):
cur_node = cur_node.next
new_head = cur_node.next
cur_node.next = None
return new_head
参考文献:
https://leetcode-cn.com/problems/rotate-list/solution/xuan-zhuan-lian-biao-by-leetcode-solutio-woq1/