给定一个链表,旋转链表,将链表每个节点向右移动 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
题目标签:Linked List / Two Pointers
题目链接:LeetCode / LeetCode中国
比较简单的单链表题目,注意空链表!
Language | Runtime | Memory |
---|---|---|
python3 | 48 ms | N/A |
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def rotateRight(self, head, k):
"""
:type head: ListNode
:type k: int
:rtype: ListNode
"""
nodes = []
while head:
nodes.append(head)
head = head.next
if nodes:
k %= len(nodes)
if k > 0:
nodes[-1].next = nodes[0]
nodes[len(nodes) - k - 1].next = None
return nodes[-1 * k]
else:
return None