递归问题一直是一个看似简单实则上手很难的方法。很多人(包括我在内)常常无法把具体问题抽象成递归函数,找不到终止条件,不知道函数应该做什么,不清楚要抽象出的逻辑什么...
本文尝试用leetcode中的简单链表问题,来一步步阐述递归的思考过程。
首先要明白,递归是反复调用自身的过程。递归分为递推和回归两步,递推和回归的中间过渡为终止条件。我们不必考虑每一步递归究竟在做什么,只需要把问题上升一个高度,不去关注细节,只关注宏观逻辑。即可知递归一大关键要素:抽象。
题目:
思考如何递归往往要考虑三个条件:
- 终止条件是什么?
- 本层做什么?
- 返回什么值?
根据题目,终止条件是当当前节点为null或当前节点的next为null。
要回答本层做什么,首先要把链表抽象出来,忽略其中细节。
反转链表,即当前节点和下一节点的指向顺序改变:
cur.next.next = cur
所以,归根结底每一层递归只需考虑当前节点和下一节点即可。
也可理解为:
假设链表为:
n1 ->... ->nk-1 -> nk -> nk+1 -> ... -> nm -> null
若从节点nk+1到nm已经被反转,而我们正处于nk。
n1 -> ... ->nk-1 -> nk -> nk+1 <- ... <- nm
我们希望nk+1的下一个节点指向nk
所以,nk.next.next = nk。
由于n1要指向null,所以nk.next = null
最后一个要素:返回值是什么?
个人认为这是本题的难点,首先我们考虑最后一次回归的返回值是什么?是反转链表的头节点,即原链表的最后一个节点(终止条件)。
以1->2->3->4->5->null为例,最后一次回归的返回值应是指向5的头节点,所以得到的答案为:5->4->3->2->1
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def reverseList(self, head: ListNode) -> ListNode:
if head == None or head.next == None:
return head
temp = self.reverseList(head.next)
head.next.next = head
head.next = None
return temp
如何理解返回值temp?
temp为满足终止条件时返回的head,即每次返回都是同一个值(原链表最后一个节点,即反转后的头节点)。
方法二:
抽象为pre节点和cur节点。
class Solution:
def reverseList(self, head: ListNode) -> ListNode:
return self.reverse(None, head)
def reverse(self, pre: ListNode, cur: ListNode) -> ListNode:
if cur == None:
return pre
next_ = cur.next
cur.next = pre
return self.reverse(cur, next_)