leetcode hot100 之 反转链表

题目

给定一个链表，请将其返回再返回头节点。

输入：head = [1,2,3,4,5]
输出：[5,4,3,2,1]

原题链接：https://leetcode.cn/problems/reverse-linked-list/

思路

维护一个 pre 和 cur 以及 next 指针，不断地使 cur 指向 pre，然后更新 pre 、cur 和 next。
利用 preHead 更方便。
最终记得处理下 head，使其指向NULL。

• 复杂度分析
  • 时间复杂度 O(n)。遍历所有节点。
  • 空间复杂度 O(1)。

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* reverseList(ListNode* head) {
        if (head == NULL) {
            return head;
        }
        ListNode* preHead = new ListNode();
        preHead->next = head;
        ListNode* pre = preHead;
        ListNode* cur = preHead->next;
        while (cur) {
            ListNode* next = cur->next;
            cur->next = pre;
            pre = cur;
            cur = next;
        }
        head->next = NULL;
        return pre;
    }
};