1. 题目描述
- 旋转链表
给定一个链表,旋转链表,将链表每个节点向右移动 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
2. 代码实现
方法:先将链表形成一个闭环,找到最后一个节点和首节点位置,形成一个新的单链表。
执行用时 :32 ms, 在所有 Python 提交中击败了40.30%的用户
内存消耗 :12.8 MB, 在所有 Python 提交中击败了11.11%的用户
# Definition for singly-linked list.
class ListNode(object):
def __init__(self, x):
self.val = x
self.next = None
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
current = head
lens = 1
while current.next:
current = current.next
lens += 1
# 形成闭环
current.next = head
temp = head
lastnode = lens - k%lens-1
# 获得最后一个节点
for i in range(lastnode):
temp = temp.next
newhead = temp.next
temp.next = None
return newhead