March——61.旋转链表

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def rotateRight(self, head: ListNode, k: int) -> ListNode:
        if not head or not head.next or not k: return head
        tmp, count= head, 0
        #先计算链表总长度
        while tmp:
            tmp = tmp.next
            count += 1
        #取余
        k = k % count

        #如果k=0不用反转
        if k == 0:
            return head
        
        #设置快慢指针,让快指针比慢指针快走k步
        fast = slow = head
        for i in range(k):
            fast = fast.next
        
        #然后快慢指针起步走,这样就能保证找到倒数的k个节点
        while fast.next:
            fast = fast.next
            slow = slow.next
        
        #断掉链表
        newHead = ListNode(0)
        newHead = slow.next
        
        #让原链表的下个节点为None
        slow.next = None
        #让快指针指向head即可
        fast.next = head
        return newHead
  • 基本思路
    • 先计算链表的总长度,然后k对总长度进行取余
    • 如果取于的结果为0,那就直接放回
    • 否则,设置快慢指针,让快指针先走k步
    • 然后快慢指针同步走,直到快指针走到链表尾部
    • 设置一个新的节点newHead,该节点就等于慢指针的下个节点
    • 然后让慢指针的下个节点为空,断开链表,并且让快指针的的下个指针指向链表的头结点
    • 然后返回新的节点,这个就是翻转后的新链表 

其实本质就是翻转后k个元素,然后和前面的链表调换顺序,fuxuemingzhu的这个图话的不错。fuxuemingzhu

猜你喜欢

转载自blog.csdn.net/weixin_37724529/article/details/115262379