AcWing 35. 反转链表

问题描述：

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

样例

输入: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) {
        if(!head) return nullptr;
        ListNode *h = new ListNode(head -> val), *p;
        head = head -> next;
        while(head){
            p = new ListNode(head -> val);
            head = head -> next;
            p -> next = h;
            h = p;
        }
        return h;
    }
};

看过y总视频之后受到启发，发现可以保存该节点的上一个节点，然后修改next指向就行
空间复杂度：O（1）

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