LeetCode 1721 交换链表中的节点

题目

LeetCode 1721 交换链表中的节点

给你链表的头节点 head 和一个整数 k。

交换 链表正数第 k 个节点和倒数第 k 个节点的值后，返回链表的头节点（链表 从 1 开始索引）。

示例 1：

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

示例 2：

输入： head = [7,9,6,6,7,8,3,0,9,5], k = 5 输出： [7,9,6,6,8,7,3,0,9,5]

示例 3：

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

提示：

• 链表中节点的数目是 n
• 1 <= k <= n <= 10^5
• 0 <= Node.val <= 100

代码实现

C++

/**
 * 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* swapNodes(ListNode* head, int k) {
        // 1. 找到正数第 k 个节点
        ListNode* first = head;
        for (int i = 0; i < k - 1; ++i) {
            first = first->next;
        }
        
        // 2. 找到倒数第 k 个节点（快慢指针法）
        ListNode* second = head;
        ListNode* fast = first; // 快指针先站在正数第 k 个节点，形成 k 步间隔
        while (fast->next != nullptr) {
            fast = fast->next;
            second = second->next;
        }
        
        // 3. 仅交换两个节点的值
        std::(first->val, second->val);
         head;
    }
};