题目
输入一个链表,反转链表后,输出新链表的表头。
思路
迭代法:
画图模拟,发现需要三个指针来完成反转,需要保存下一个节点,因为反转后会丢失和后面节点的连接。注意头节点的next要变成None(尾节点),返回节点是原来的尾节点。
递归法:
新建一个头节点指向尾节点的,每次递归返回值不变。每次递归传入的节点pHead是此次要加入反转链表的节点,pHead.next指向的是已经反转过的链表的最后一个节点,所以把pHead赋值给pHead.next.next,成功反转该节点。不要忘记把pHead.next设为None,结束链表,尾节点。
代码
def ReverseList(pHead):
if pHead == None or pHead.next == None:
return pHead
pahead = pHead
p = pHead.next
pafter = p.next
pHead.next = None # 头节点变尾节点
while pafter != None:
p.next = pahead
pahead = p
p = pafter
pafter = pafter.next
p.next = pahead
return p
def reverseList(self, head: ListNode) -> ListNode:
pre = None
cur = head
while cur != None:
ne = cur.next
cur.next = pre
pre = cur
cur = ne
return pre
def ReverseList(pHead):
if pHead == None or pHead.next == None:
return pHead
else:
newHead = ListNode(0)
newHead = ReverseList(pHead.next)
pHead.next.next = pHead
pHead.next = None
return newHead
进阶
对第m-n个节点反转,其余节点不变
虚头节点便于m=1的情况下的反转
- 找到要反转的前一个节点pre
- 找到第一个要反转的节点cur
- 找到当前要反转的节点ne
- 反转后的最后一个节点,与未反转的节点连接
- 当前反转节点作为反转后的头节点,next指向之前的头节点
- 更新反转后的头节点
(这里的头节点指待反转链表的头节点,不是整个链表的头节点)
def reverseBetween(self, head: ListNode, m: int, n: int) -> ListNode:
if head == None:
return None
dummyhead = ListNode(0)
dummyhead.next = head
pre = dummyhead
for _ in range(m-1):
pre = pre.next # pre指向反转的前一个节点,用来连接待反转的最后一个节点
cur = pre.next # cur指向第一个要反转的节点,是反转后的最后一个节点
for _ in range(n-m):
ne = cur.next #保留下一个节点信息,ne是此次要和cur反转的节点
cur.next = ne.next #反转后的最后一个节点,与未反转的节点连接
ne.next = pre.next #注意这里不能直接等于cur,因为cur不变,一直指向第一个要反转的节点,所以要通过pre.next去找反转后的头节点
pre.next = ne #更新反转后的头节点
return dummyhead.next