反转链表
描述
给定一个单链表的头结点pHead(该头节点是有值的,比如在下图,它的val是1),长度为n,反转该链表后,返回新链表的表头。
数据范围: 0\leq n\leq10000≤n≤1000
要求:空间复杂度 O(1)O(1) ,时间复杂度 O(n)O(n) 。
如当输入链表{1,2,3}时,
经反转后,原链表变为{3,2,1},所以对应的输出为{3,2,1}。
示例1
输入:
{1,2,3}
返回值:
{3,2,1}
示例2
输入:
{}
返回值:
{}
说明:
空链表则输出空
解决方案
反转链表跟从尾到头打印链表差不多
反转链表需要将指向换一下,本来每个节点指向他的下一个节点,现在变成了他的下一个节点指向他
思路一:做一个新链表
def ReverseList(self , head: ListNode) -> ListNode:
# write code here
if head == None or head.next == None:
return head
pre = None
cur = head
while cur != None:
temp = cur.next
cur.next = pre
pre = cur
cur = temp
return pre
思路一变形,使用递归
结束条件就是当头指针尾空或者链表中只有一个元素时,返回头指针,递归是参考的别人的代码,原来还可以这样写,学到了
def ReverseList(self , head: ListNode) -> ListNode:
# write code here
def reverse(pre, cur):
if not cur:
return pre
tmp = cur.next
cur.next = pre
return reverse(cur, tmp)
return reverse(None, head)
参考资料
牛客网评论区