206反转链表

反转链表

/**
 * 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 == nullptr || head->next == nullptr) {
            return head; // 如果链表为空或只有一个节点，直接返回head
        }

        ListNode* prev = nullptr; // 前一个节点
        ListNode* current = head; // 当前节点
        ListNode* next = nullptr; // 下一个节点

        while (current != nullptr) {
            next = current->next; // 保存下一个节点
            current->next = prev; // 反转当前节点的指针
            prev = current;       // 前一个节点移动到当前节点
            current = next;       // 当前节点移动到下一个节点
        }

        head = prev; // 新的头节点是prev
        return head;
    }
};