题目:定义一个函数,输入一个链表的头结点,反转该链表并输出反转后链表的头结点
数据范围:链表长度 [0,30]。
**样例**
输入:1->2->3->4->5->NULL
输出:5->4->3->2->1->NULL
算法一:迭代O(n) 双指针
翻转即将所有节点的next指针指向前驱节点。由于是单链表,我们在迭代时不能直接找到前驱节点,所以我们需要一个额外的指针保存前驱节点。同时在改变当前节点的next指针前,不要忘记保存它的后继节点,以便下一个结点的翻转。
//迭代遍历只需遍历一遍所以时间复杂度为O(n),只用到了三个额外的指针所以空间复杂度为O(1)
struct ListNode {
int val;
ListNode* next;
ListNode(int x) : val(x), next(NULL) {}
};
class Solution {
public:
ListNode* reverseList(ListNode* head) {
if (!head || head->next == nullptr)//空链表或链表中只有一个结点则直接返回头结点
return head;
ListNode* pre = head;
ListNode* cur = head->next;
while (cur)//迭代,不断地让结点的next指向它的前驱结点
{
auto next = cur->next;
cur->next = pre;
pre = cur;
cur = next;
}
head->next = nullptr;
return pre;
}
};
int main()
{
ListNode* l1 = new ListNode(1);
ListNode* l2 = new ListNode(2);
ListNode* l3 = new ListNode(3);
l1->next = l2;
l2->next = l3;
//1->2->3->null
Solution s;
cout << s.reverseList(l1)->val << endl;//输出3
//3->2->1->null
}
算法二:递归O(n)
根据递归的思想,要想翻转1->2->3->null链表,需要先翻转2->3->null链表;要想翻转2->3->null链表,需要先翻转3->null链表。
首先我们先考虑 reverseList 函数能做什么,它可以翻转一个链表,并返回新链表的头节点,也就是原链表的尾节点。所以我们可以先递归处理 reverseList(head->next),这样我们可以将以head->next为头节点的链表进行翻转,我们令当前头结点的后继结点的next指针指向head,并将head->next指向空即可将当前新链表链表翻转。最后返回当前新链表的头结点,不断递归直到把整个链表翻转。
空间复杂度分析:总共递归 n 层,系统栈的空间复杂度是O(n),所以总共需要额外O(n)的空间。
时间复杂度分析:链表中每个节点只被遍历一次,所以时间复杂度是 O(n)
class Solution {
public:
ListNode* reverseList(ListNode* head) {
if(!head || head->next==nullptr)//递归结束边界值,为了找到原链表的尾节点
return head;
ListNode* tail=reverseList(head->next);//接收链表的尾结点(翻转后即称为新链表的头结点)
head->next->next=head;//将当前链表的头结点的后继结点的next指向头结点进行翻转
head->next=nullptr;//翻转后,当前头结点就成为了新链表的尾节点,需要将尾节点的next指向空
return tail;//返回新链表的头结点
}
};