力扣24-两两交换链表中的节点-C++

一、题目

给你一个链表，两两交换其中相邻的节点，并返回交换后链表的头节点。你必须在不修改节点内部的值的情况下完成本题（即，只能进行节点交换）。

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

输入：head = []
输出：[]
示例 3：

输入：head = [1]
输出：[1]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/swap-nodes-in-pairs

二、思路

从前往后扫描链表，每次对两个链表节点进行交换，若当前节点时最后一个节点或只剩下一个节点，则可以返回新链表，这里需要注意的是新链表的头结点是原链表的第二个结点（原结点数超过2时），在交换时对头结点需要处理一下。

三、代码

class Solution {
public:
    ListNode* swapPairs(ListNode* head) {
        if(head == NULL || head->next == NULL) return head;
        ListNode* p = head;
        head = head->next; //head指向第二个结点，交换后就是新的结点
        p->next = head->next;
        head->next = p;
        ListNode* pre = p;  //指向p的前驱结点（已交换部分的最后一个节点）
        p = p->next;
        ListNode* q; //指向p的后继结点
        while(p != NULL && p->next != NULL){
            q = p->next;  //两两交换
            p->next = q->next;
            pre->next = q;

            q->next = p;  //pre, p后移
            pre = p;
            p = p->next;
        }
        return head;
    }
};

四、运行结果