原题:
给定一个链表,旋转链表,将链表每个节点向右移动 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/rotate-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解法一:
思路:这个题目存在一个规律。当我们向移动K个位置的时候,其实就是将链表前(length - k)个节点整体移接到链表的最后。
下面这个程序就是这个思想,找出第length - k的位置,直接将链表切成两段再拼接。
程序一
# 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 head == None:
return None
pre = ListNode(0)
pre.next = head
length = 1
ans = head
while ans.next != None:
ans = ans.next
length += 1
k = k % length
if k == length or k == 0:
return head
for i in range(length - k):
pre =pre.next
tail = new_head = pre.next
pre.next = None
while tail.next != None:
tail = tail.next
tail.next = head
head = new_head
return head
程序二:
程序二相对程序一,最大的不同是将链表变为了循环链表,所以在链表的分割上会更为便捷一些。
# 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 head == None or k==0 or head.next == None:return head
length = 0
temp = head
while temp:
temp = temp.next
length +=1
k = k % length
temp = head
for i in range(length-1):
temp = temp.next
temp.next = head # 形成了一个闭环
temp = head
for i in range(length -k):
temp = temp.next
head = temp
for i in range(length-1):
temp = temp.next
temp.next = None
return head