剑指Offer系列——24.反转链表

题目

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

示例

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

限制

0 <= 节点个数 <= 5000

思路

我们用两个指针指向最开始的两个节点，操作为将箭头反转，然后两个指针后移，重复操作直到头。注意判空

代码

/**
 * 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 NULL;
        auto  a = head, b = head->next;
        a->next = NULL;
        while (b) {
            auto c = b->next;
            b->next = a;
            a = b;
            b = c;
        }
        return a;
    }
};