24. 反转链表

原题链接

题目描述

定义一个函数，输入一个链表的头节点，反转该链表并输出反转后链表的头节点。

示例:

输入: 1->2->3->4->5->NULL
输出: 5->4->3->2->1->NULL

限制：

\(0\) \(≤\) 节点个数 \(≤\) \(5000\)

---

遍历

时间复杂度\(O(n)\)  空间复杂度\(O(1)\)

定义前驱结点为pre，当前结点为cur，由于每次反转需要把当前结点的next指针指向前驱结点pre，所以需要再定义一个next用来存储当前结点的下一个结点。

反转操作：

1. 把当前结点的next指针指向前驱结点；
2. 把前驱结点pre移到当前结点cur上；
3. 把当前结点cur移到之前定义的next结点处。

C++

/**
 * 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 = nullptr;
        ListNode* cur = head;
        while (cur) {
            ListNode* next = cur->next;
            cur->next = pre;
            pre = cur;
            cur = next;
        }
        return pre;
    }
};

递归

时间复杂度\(O(n)\)  空间复杂度\(O(n)\)

定义tail结点为reverseList(head)函数的返回值。当没有next结点时，返回head结点。

否则，将当前结点的next结点的next指向当前head结点，head->next置为空指针nullptr。

C++

/**
 * 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) {
        if (!head || !head->next) return head;

        ListNode* tail = reverseList(head->next);
        head->next->next = head;
        head->next = nullptr;

        return tail;
    }
};