定义一个函数,输入一个链表的头结点,反转该链表并输出反转后链表的头结点。
思考题:
请同时实现迭代版本和递归版本。
样例
输入:1->2->3->4->5->NULL
输出:5->4->3->2->1->NULL
代码
解法一 迭代,O(n)
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* reverseList(ListNode* head) {
ListNode* pre=NULL,*cur=head,*next=NULL;
while(cur){
next=cur->next;//记录当前结点的后继结点
cur->next=pre;//当前结点的next指向他原来的前趋结点,即翻转当前结点
pre=cur;//翻转后当前结点为翻转后链表的头结点
cur=next;//当前结点指向未翻转链表的头结点
}
return pre;//不管什么时候pre都是指向翻转后链表的头结点
}
};
解法二 递归,O(n)
图1
class Solution {
public:
ListNode* reverseList(ListNode* head) {
//空链表或者是只有一个结点时直接返回即可无需反转
if(!head || !head->next)
return head;
/*注意递归出口点:
如样例的1->2->3->4->5->NULL,当head->next指向5时(当前head指向4),调用reverseList函数直接返回5
并赋值为当前head指向4的那次调用的tail指针,tail指向的内容为5->NULL。然后按照下面的步骤反转链表
得到5->4->NULL,其他的以此类推。
*/
//reverseList返回的是反转后的链表头结点(即原链表的尾结点)
ListNode* tail = reverseList(head->next);
//因为没有修改过head指针的next值,所以它还是指向原链表的下一个节点(即反转后的链表尾结点)
//让反转后的链表尾结点指向自己,即反转当前结点(head指向的那个结点)
//如样例的1->2->3->4->5->NULL,当head指向1时,返回的是5->4->3->2->NULL,如图1所示
//因为head->next原来是指向2的且没有改变过,所以head->next->next = head的含义为让1成为2的后继结点
//即反转当前结点1
head->next->next = head;
head->next=NULL;//每次调用reverseList后当前结点为尾结点所以next值要赋为NULL
return tail;//返回反转后的链表头结点(即本次调用时原链表的尾结点)
}
};