看到题第一反应,是借助栈,先进后出,直接实现链表反转,但是空间复杂度是O(n),较为浪费空间,所以才会进行下面的两种优化方式1.迭代 2.递归——两种方法空间复杂度都是O(1)
一、迭代
还是借用双指针的思想,刚开始没有想清楚的是如果只是按照链表顺序从head调换一个箭头的正反,那链条不就是断了吗,怎么找下一个node?但其实pre指针一直记录着下一个指针迭代位置,所以不会丢失数据,这点通了之后,迭代就很好理解了
class Solution:
def reverseList(self, head: Optional[ListNode]) -> Optional[ListNode]:
pre,cur=head,None
while pre:
tmp=pre.next#借助tmp指针记录下一个结点
pre.next=cur
#去进行下一轮迭代
cur=pre
pre=tmp
return cur
class Solution {
public:
ListNode* reverseList(ListNode* head) {
ListNode *pre=head,*cur=nullptr;
ListNode *tmp;
while(pre){
tmp=pre->next;
pre->next=cur;
cur=pre;
pre=tmp;
}
return cur;
}
};
二、递归
递归出口:当cur指针为空的时候
class Solution:
def reverseList(self, head: Optional[ListNode]) -> Optional[ListNode]:
def recur(cur,pre):
if not cur: return pre
res=recur(cur.next,cur)#递归后继结点
cur.next=pre#修改结点方向
return res#返回反转链表的头结点
return recur(head,None)#递归调用并返回
class Solution {
public:
ListNode* reverseList(ListNode* head) {
return recur(head,nullptr);
}
private:
ListNode* recur(ListNode* pre,ListNode* cur){
if(pre==nullptr){
return cur;
}
ListNode *res=recur(pre->next,pre);//递归直至最后一个结点
pre->next=cur;//反转链表
return res;
}
};
小tips:
1、如果空间复杂度不为1的时候,一定要多想一想有没有优化的方法,尽可能优化
2、如果问题可以用迭代解决的时候,一定要想一下是否能用递归