24. 两两交换链表中的节点

题目描述

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

示例 1：

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

示例 2：

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

示例 3：

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

提示：

• 链表中节点的数目在范围 [0, 100] 内
• 0 <= Node.val <= 100

解答

/**
 * 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* swapPairs1(ListNode* head) {
        // 先把前面两个元素对调，然后再对后面的链表进行相同处理
        if(head == nullptr || head->next == nullptr)
        {
           return head;
        }

        ListNode *pre = head;
        ListNode *cur = head->next; // 第二个节点
        ListNode *nxt = head->next->next; // 第三个节点
        pre->next = swapPairs(nxt);
        cur->next = pre;
        return cur;
    }

    // 迭代法
    ListNode* swapPairs(ListNode* head) {
        ListNode *dummyHead = new ListNode(0); //
        dummyHead->next = head;
        ListNode *cur = dummyHead;

        while(cur->next != nullptr && cur->next->next != nullptr)
        {
            ListNode *tmp = cur->next; // 存第一个元素
            ListNode *tmp1 = cur->next->next->next; // 下一组对换的开头

            cur->next = cur->next->next; // 指向第二个元素
            cur->next->next = tmp;  // 原来第二个元素指向第一个元素
            cur->next->next->next = tmp1; // 原来第一个元素指向 下一组对换的开头

            cur = cur->next->next;
        }
        return dummyHead->next;
    }
};