代码
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
### 0208 遍历(40 ms,14.8 MB)
def rotateRight(self, head: ListNode, k: int) -> ListNode:
# 注意:当链表只有一个元素时,需要返回头节点head,而非空!
if not head or not head.next: return head
# 注意:由于需要闭合链表,因此链表长度需要初始化为1!
dum, length = head, 1
# 注意:需判断节点的next指针是否为空,而非直接判断节点是否为空,因为需要使用next指针来闭合链表!
while dum.next:
dum = dum.next
length += 1
# 闭合链表(链表成环)
dum.next = head
# 注意:由于k可能大于链表长度,所以需要先对链表长度求余!
k %= length
# 注意:链表右移k,相当于将位于length-k的节点作为新的头节点!
for _ in range(length-k):
dum = dum.next
# 注意:新的头节点为length-k处的节点,这里需要使用next来获取!
head = dum.next
# 注意:需要断开环链表!
dum.next = None
return head