反转一个单链表。
示例:
输入: 1->2->3->4->5->NULL
输出: 5->4->3->2->1->NULL
这道题我刚开始以为需要在原来的链表上修改,然后有了下面的代码,结果超时了。看题解才发现是可以返回值的。
class Solution:
def reverseList(self, head: ListNode) -> ListNode:
length = 0
p = head
while p:
p = p.next
length += 1
p = head
i = 0
while p:
q = p
i += 1
j = length - i
while q:
if j > 0:
q = q.next
j -= 1
p.val, q.val = q.val, p.val
p = p.next
迭代法(向前考虑)
假如第i个元素之前都已经翻转了,第i个元素该如何做
- 首先获得第i+1个元素的指针, p.next保存为temp
- 第i的元素的next应该赋值为res, res是翻转的部分
- res要始终指向最前面的元素,所以res前移
- 把temp重新赋值给p
class Solution:
def reverseList(self, head: ListNode) -> ListNode:
res = None
p = head
while p:
temp = p.next
p.next = res
res = p
p = temp
return res
比如说有三个数字的链表。
1>2>3>None
第一次迭代之后
p = 2->3>None
res = 1>None
第二次迭代之后
p = 3->None
res = 2>1>None
…
递归法
递归法则是向后考虑,假设第i个元素之后的序列都翻转好了,第i个元素该怎么做呢
- 此时, 序列满足 …, i > i+1 < i+2 < … i+M
- 那么i的next就是i+1,让i的next的next为i
- 然后i的next为None, 此时序列为 … i-1 > i < i+1 < … i+M
- 返回指向最后元素的指针res。res始终指向最后一个元素。迭代法res始终指向第一个元素。
class Solution:
def reverseList(self, head: ListNode) -> ListNode:
if head == None or head.next == None:
return head
res = self.reverseList(head.next)
head.next.next = head
head.next = None
return res