Leetcode刷题记录 61. 旋转链表

 

给定一个链表，旋转链表，将链表每个节点向右移动 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

（1）遍历链表求长度；

（2）cur指针向前移动k次；

（3）pre和cur指针向前移动直到尾部；

（4）把pre的下一节点作为头结点，原有的头结点接到尾部，pre节点下一节点赋空。

class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None

class Solution:
    def rotateright(self, head:ListNode, k: int):
        pre=head
        lennum=1
        if ( head==None):return head
        while(pre.next):
            pre=pre.next
            lennum+=1    
        pre=head
        k=k%lennum
        if(k==0 or head.next==None):
            return head
        cur=head.next
        while(k-1>0):
            cur=cur.next
            k-=1
        while(cur.next!=None):
            pre=pre.next
            cur=cur.next
        res=pre.next
        pre.next=None
        cur.next=head
        return res

if __name__=="__main__":
    head=ListNode(1)
    last=head
    for i in[2,3]:
        last.next=ListNode(i)
        last=last.next
    print(Solution().rotateright(head,3))