leetcode-反转链表II(python)

题目:
反转从位置 m 到 n 的链表。请使用一趟扫描完成反转。

说明:

1 ≤ m ≤ n ≤ 链表长度。

示例:

输入: 1->2->3->4->5->NULL, m = 2, n = 4
输出: 1->4->3->2->5->NULL

代码:

class Solution:
    def reverseBetween(self, head, m, n):
        changelen=n-m+1  #计算需要逆置的节点个数
        result=head  #记录头节点
        prehead=None #记录开始逆转点的前一个节点
        while m-1:  #找到逆转点位置
            prehead=head  #逆转点的前一个
            head=head.next #逆转点
            m-=1
        tail=head #逆序后要指向链表尾部
        newhead=None  #逆转过程
        while changelen:
            nex=head.next
            head.next=newhead
            newhead=head
            head=nex
            changelen-=1  
        if prehead:  #如果不为空,说明不是从第一个点开始逆转
            prehead.next=newhead
        else:result=newhead #从第一个点开始逆转的情况
        tail.next=head
        return result

猜你喜欢

转载自blog.csdn.net/weixin_40876685/article/details/88722800