LeetCode: 两两交换链表中的节点 python实现

1. 题目描述

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

2. 代码实现

        方法:先将链表形成一个闭环,找到最后一个节点和首节点位置,形成一个新的单链表。
        执行用时 :32 ms, 在所有 Python 提交中击败了40.30%的用户
        内存消耗 :12.8 MB, 在所有 Python 提交中击败了11.11%的用户

# 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 not  head or not head.next:
            return head
        current = head
        lens = 1
        while current.next:
            current = current.next
            lens += 1
        # 形成闭环
        current.next = head
        temp = head
        lastnode = lens - k%lens-1
        # 获得最后一个节点
        for i in range(lastnode):
            temp = temp.next
        newhead = temp.next
        temp.next = None

        return newhead
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章