35. 反转链表

文章目录

• ​​Question​​
• ​​Ideas​​
• ​​Code​​

• ​​迭代版​​
• ​​递归版​​

Question

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

思考题：

请同时实现迭代版本和递归版本。
数据范围
链表长度 [0,30]。

样例
输入:1->2->3->4->5->NULL

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

Ideas

Code

迭代版

/**
 * 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;
        
        auto a = head, b = head -> next;
        // 指针反转并且a和b两个指针后移一位
        while (b)
        {
            auto c = b -> next;
            b -> next = a;
            a = b;
            b = c;
        }
        head -> next = NULL;

        return a;
    }
};

递归版

/**
 * 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;

        auto tail = reverseList(head -> next);
        head -> next -> next = head; // head -> next 第二个节点
        head -> next = NULL;

        return tail;
    }
};