题目: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