首页 > 其他 > 详细

LeetCode 61. 旋转链表

时间:2020-07-12 11:30:41      阅读:65      评论:0      收藏:0      [点我收藏+]

给定一个链表,旋转链表,将链表每个节点向右移动?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

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def rotateRight(self, head: ListNode, k: int) -> ListNode:
        if head is None or head.next is None :
            return head 
        cnt = 1 
        tmp = head 
        while tmp.next is not None:
            tmp = tmp.next 
            cnt += 1
        k = k%cnt 
        #print(cnt,k)
        if k==0:
            return head 
        tail1 = head 
        tail2 = head 
        for i in range(k):
            tail2 = tail2.next 
        while tail2.next is not None:
            tail1 = tail1.next 
            tail2 = tail2.next
        #print(head.val,tail1.val,tail2.val)
        tail2.next = head  
        head = tail1.next 
        tail1.next = None 
        return head

LeetCode 61. 旋转链表

原文:https://www.cnblogs.com/sandy-t/p/13287268.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!