2.4 剑指offer 反转链表

题目

输入一个链表,反转链表后,输出新链表的表头。

思路

迭代法:

画图模拟,发现需要三个指针来完成反转,需要保存下一个节点,因为反转后会丢失和后面节点的连接。注意头节点的next要变成None(尾节点),返回节点是原来的尾节点。

递归法:

2.4 剑指offer 反转链表_第1张图片
2.4 剑指offer 反转链表_第2张图片
新建一个头节点指向尾节点的,每次递归返回值不变。每次递归传入的节点pHead是此次要加入反转链表的节点,pHead.next指向的是已经反转过的链表的最后一个节点,所以把pHead赋值给pHead.next.next,成功反转该节点。不要忘记把pHead.next设为None,结束链表,尾节点。
2.4 剑指offer 反转链表_第3张图片

代码

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的情况下的反转

  1. 找到要反转的前一个节点pre
  2. 找到第一个要反转的节点cur
  3. 找到当前要反转的节点ne
  4. 反转后的最后一个节点,与未反转的节点连接
  5. 当前反转节点作为反转后的头节点,next指向之前的头节点
  6. 更新反转后的头节点
    (这里的头节点指待反转链表的头节点,不是整个链表的头节点)
 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

leetcode 题解

2.4 剑指offer 反转链表_第4张图片

你可能感兴趣的:(刷题)